Softmax Dim parameter


So say I have a 3D tensor [Channels x Width x Height] and I would like to perform softmax along the channel dimension. Do I specify the parameter dim=0 or dim=2?

Thank you


We read dimensionality left to right, so in your case it have to be dim=0.

Good luck

1 Like