Mojo module

softmax

Functions

• ​identity:
• ​logsoftmax:
• ​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:
• ​softmax_with_temperature: GPU softmax with per-row temperature scaling.
• ​sub:

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit