Class

MPSMatrixSoftMax

A softmax kernel that operates on matrices.

Declaration

@interface MPSMatrixSoftMax : MPSMatrixUnaryKernel

Relationships

See Also

Matrix Softmax Operations

MPSMatrixLogSoftMax

A logarithmic softmax kernel that operates on matrices.

MPSMatrixLogSoftMaxGradient

A logarithmic gradient softmax kernel that operates on matrices.

MPSMatrixSoftMaxGradient

A gradient softmax kernel that operates on matrices.