Class

MPSMatrixSoftMaxGradient

A gradient softmax kernel that operates on matrices.

Declaration

class MPSMatrixSoftMaxGradient : MPSMatrixBinaryKernel

Topics

Initializers

init?(coder: NSCoder, device: MTLDevice)
init(device: MTLDevice)

Instance Properties

var sourceColumns: Int
var sourceRows: Int

Instance Methods

func copy(with: NSZone?, device: MTLDevice?) -> Self
func encode(to: MTLCommandBuffer, gradientMatrix: MPSMatrix, forwardOutputMatrix: MPSMatrix, resultMatrix: MPSMatrix)

Relationships

Inherits From

Conforms To

See Also

Matrix Softmax Operations

class MPSMatrixLogSoftMax

A logarithmic softmax kernel that operates on matrices.

class MPSMatrixLogSoftMaxGradient

A logarithmic gradient softmax kernel that operates on matrices.

class MPSMatrixSoftMax

A softmax kernel that operates on matrices.