Mojo module
softmax
Functions
-
identity
: -
logsoftmax
: Performs an unbatched logsoftmax on an input tensor using the three-pass algorithm. -
mul
: -
reciprocal
: -
reduce_add_simd
: This functions adds val to either the scalar value or the vector value depending on the step_simd_width. This is useful when the simd_width varies between iterations as in vectorize. -
softmax
: -
softmax_2_pass
: Performs an unbatched softmax on an input tensor using the two-pass online algorithm. -
softmax_3_pass
: Performs an unbatched softmax on an input tensor using the three-pass algorithm. -
softmax_kernel
: -
sub
:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!