I have a matrix X of shape (N,D) ans a vector mu of shape(D,K). I tried subtracting one column of mu from one row of X and am getting the wrong answer. When I convert it to numpy array, I am getting the right answer. What am I doing wrong?
print(X[0,:])
print(mu[:,0])
print(X[0,:]-mu[:, 0])
print(X[0,:].data.numpy()-mu[:, 0].data.numpy())
tensor([-2.4682,  0.9872, -0.4661,  1.7443,  0.5375, -0.5948,  2.2457, -0.2980,
1.2467,  0.2932], dtype=torch.float64)
tensor([-1.5256, -0.6092, -0.7773,  0.4676,  0.8657, -0.1759, -0.7981, -0.4370,
0.2152, -0.8696], grad_fn=)
tensor([ -9.4257e-01,   1.1631e+00,  -4.6614e-01,   1.7443e+00,   5.3751e-01,
-5.9480e-01,   2.2457e+00, -1.1146e+171,   2.1892e+00,   8.8797e-01],
dtype=torch.float64, grad_fn=)
[-0.94256972  1.59635256  0.31117838  1.27669038 -0.32818301 -0.41886666
3.0437956   0.13902091  1.03149757  1.16275023]