Contents

softMaxCrossEntropyGradient(_:source:labels:axis:reuctionType:name:)

Creates the gradient of a softmax cross-entropy loss operation and returns the result tensor.

Declaration

func softMaxCrossEntropyGradient(_ gradientTensor: MPSGraphTensor, source sourceTensor: MPSGraphTensor, labels labelsTensor: MPSGraphTensor, axis: Int, reuctionType reductionType: MPSGraphLossReductionType, name: String?) -> MPSGraphTensor

Parameters

  • gradientTensor:

    The input gradientTensor. Note: in most cases this is the initial gradient tensor, which is a constant tensor with value one.

  • sourceTensor:

    The source tensor.

  • labelsTensor:

    The labels tensor.

  • axis:

    The axis over which the operation computes the softmax reduction.

  • reductionType:

    The type of reduction MPSGraph uses to reduce across all other axes than axis. See: Mpsgraphlossreductiontype.

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor object.