This documentation is from a previous release and may not reflect the latest released version. View latest release documentation.
Skip Navigation
Class

SoftMax

Applies the Softmax function.
class SoftMax
Activations.swift

Overview

This is:

MLX.softmax(x)

See Also

Topics

Instance Methods

func callAsFunction(MLXArray) -> MLXArray

Relationships

Inherits From

Conforms To

  • MLX.Evaluatable
  • MLX.IndentedDescription
  • MLX.Updatable
  • Swift.CustomStringConvertible
  • UnaryLayer

See Also

Activation Modules

class CELU
Applies the Continuously Differentiable Exponential Linear Unit.
class GELU
Applies the Gaussian Error Linear Units function.
class GLU
Applies the gated linear unit function.
class HardSwish
Applies the hardswish function, element-wise
class LeakyReLU
Applies the Leaky Rectified Linear Unit.
class LogSigmoid
Applies the Log Sigmoid function.
class LogSoftmax
Applies the Log Softmax function.
class Mish
Applies the Mish function, element-wise.
class PReLU
Applies the element-wise parametric ReLU.
class ReLU
Applies the Rectified Linear Unit.
class ReLU6
Applies the Rectified Linear Unit 6.
class SELU
Applies the Scaled Exponential Linear Unit.
class SiLU
Applies the Sigmoid Linear Unit. Also known as Swish.
class Sigmoid
Applies the element-wise logistic sigmoid.
class Softplus
Applies the Softplus function.
class Softsign
Applies the Softsign function.
class Step
Applies the Step Activation Function.
class Tanh
Applies the hyperbolic tangent function
Current page is SoftMax