TensorPlay
API symbolsspecial
Copy
View Markdown

tensorplay.special._elementwise.softmax

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

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

On this page

Ask DeepWiki