giagrad.nn.LogSoftmax

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

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

See also

giagrad.Tensor.log_softmax()

Variables:

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