Hi,
The matrix that is shown is the Jacbian, so it’s size is always number_of_inputs x number_of_outputs so it is the right size in your case.
You should be careful that and eps of 1e-6 with FloatTensors often lead to loss of precision. I think you want to try to either use DoubleTensor (best idea usually) or increase eps.