l have a variable preds :
preds.dim()
3
[torch.cuda.FloatTensor of size 26x1x37 (GPU 0)]
l want to apply softmax on this :
m=torch.nn.Softmax()
assert input.dim() == 2, 'Softmax requires a 2D tensor as input'
AssertionError: Softmax requires a 2D tensor as input
Do l have to change the dimension of my variable ?
if yes , how can l do that
26x1x37 to become 26x37
Thank you