Softmax Dim parameter

1

Hi,

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

2

Hi,

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

Good luck

1 Like