Contents

softMaxGradient(withIncomingGradient:sourceTensor:axis:name:)

Computes the gradient of the softmax function along the specified axis using the incoming gradient tensor.

Declaration

func softMaxGradient(withIncomingGradient gradient: MPSGraphTensor, sourceTensor source: MPSGraphTensor, axis: Int, name: String?) -> MPSGraphTensor

Parameters

  • gradient:

    The incoming gradient tensor.

  • source:

    The input tensor.

  • axis:

    The axis along which softmax is computed.

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor object