Partitioned forward prop and back prop, with numerical gradient checking, note may not be correct