Hi, I am very new to pytorch and I met a runtime error when trying to do a multiplication between two tensors as following:
a = torch.nonzero(self.fake[i,0,:,:].data)
b = torch.nonzero(self.real[i, 0, :, :].data)
ab = torch.mm(a, b.t())
and the error message is :
Traceback (most recent call last):
File "train.py", line 32, in xxx
xxxx()
File "xxxx", line 311, in xxx
xxx()
File "xxxx", line 246, in xxx
ab = torch.mm(a, b.t())
RuntimeError: unimplemented data type at xxx\pytorch\torch\lib\thc\generic/THCTensorMathBlas.cu:377
Does someone know what cause this error? Thanks a lot!
The blas libraries that do matrix multiplication only support floating point numbers, so this is intentional. The error message could be better, though, I’ve opened an issue here for that: https://github.com/pytorch/pytorch/issues/4157 .