tensorplay.special._elementwise.softmax

tensorplay.special._elementwise.softmax(input, dim=None, *, dtype=None)[source]

Normalized exponential along dim (default: the last dim).