netket.nn.
softmax
Softmax function.
Computes the function which rescales elements to the range \([0, 1]\) such that the elements along axis sum to \(1\).
axis
x (Any) – input array
Any
axis (Union[int, Tuple[int, …], None]) – the axis or axes along which the softmax should be computed. The softmax output summed across these dimensions should sum to \(1\). Either an integer or a tuple of integers.
Union
int
Tuple
None