Contents

convolutionTranspose2DWeightsGradient(_:weights:outputShape:forwardConvolutionDescriptor:name:)

Creates a convolution transpose gradient operation with respect to the weights tensor of the convolution transpose operation and returns the result tensor.

Declaration

func convolutionTranspose2DWeightsGradient(_ incomingGradientTensor: MPSGraphTensor, weights source: MPSGraphTensor, outputShape: [NSNumber], forwardConvolutionDescriptor: MPSGraphConvolution2DOpDescriptor, name: String?) -> MPSGraphTensor

Parameters

  • incomingGradientTensor:

    Incoming gradient tensor

  • source:

    Forward pass source tensor

  • outputShape:

    Shape of the forward pass source weights tensor

  • forwardConvolutionDescriptor:

    Forward pass op descriptor

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor object

Discussion

Inserts an operation in graph to compute gradient of convolution transpose with respect to the weights tensor of the corresponding convolution transpose operation.