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?