Function softmax

softmax(input: Tensor, dim: number): Tensor
Computes the softmax of the input tensor along the given dimension.
Parameters
- input: Tensor
- dim: number
Returns Tensor
- Defined in functions/functional.ts:292