Mojo module
softmax
comptime values
UnsafePointer
comptime UnsafePointer = LegacyUnsafePointer[?, address_space=?, origin=?]
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: -
sub:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!