Installation
Development
High-level design
Python API (PyAsc2)
softmax()
Python API (PyAsc)
MLIR
Computes the element-wise softmax of input.
input
input – the input value (tile)
This function can also be called as a member function on Tile, as x.softmax(...) instead of softmax(x, ...).
Tile
x.softmax(...)
softmax(x, ...)