Robert

Results 1 comments of Robert

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