Hi there,
Sorry for the question, but I am still a rookie and I’m trying to understand how it works here:
I have a tensor x with shape (64,10) so 64 rows and 10 columns.
Checking shape for the below denominator I get a shape 64, why it is not equal to (64,1) and I need to reshape my tensor while using a softmax function?
torch.sum(torch.exp(x), dim=1)  -> shape 64
def softmax(x):
return torch.exp(x)/torch.sum(torch.exp(x), dim=1).view(64, 1)
Thanks a lot!