Hello, Im having a similar issue, im trying to run in gpu a torch.dot with the following code;

loss.JPG899×174 20.8 KB

but im obtaining the error;
RuntimeError: _th_dot is not implemented for type torch.cuda.LongTensor
anyone knows what should I modify to make it work ?
Thanks.