Robert
Results
1
comments of
Robert
Use torch.reshape `out = torch.bmm(weight, torch.reshape(x, (batchSize, inputSize, 1)))`