giagrad.nn.Softmax

class giagrad.nn.Softmax(*args, **kwargs)

Applies the softmax function through 1-D slices specified by axis.

See also

giagrad.Tensor.softmax()

Variables:

axis (int) – The dimension along which Softmax will be computed (so every slice along axis will sum to 1).