Hi:
I am making multiple & parallel fc layers in my network
However, Code in below image**(line 27)** causes error:
RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #2 'mat2’

What is the problem ?
Is there any better way to implement  multiple & parallel fc layers?



image.png864×367 14.5 KB