dimension

The dimension over which you want to perform the softmax operation.

var dimension: Int { get }