Preparing search index...
The search index is not available

@shumai/shumai
softmax

Function softmax

softmax(tensor: Tensor, axis: number): Tensor
Parameters
- tensor: Tensor
- axis: number
Returns Tensor
- Defined in shumai/tensor/tensor_ops.ts:10

Settings

Member Visibility

Protected
Private
Inherited
External

Theme

Modules

@shumai/shumai
- io
- loss
- module
- network
- optim
- util

Generated using TypeDoc