Contents

applyBackward(batchSize:input:output:outputGradient:generatingInputGradient:generatingBetaGradient:generatingGammaGradient:)

Applies the layer backward to generate input gradients.

Declaration

func applyBackward(batchSize: Int, input: BNNSNDArrayDescriptor, output: BNNSNDArrayDescriptor, outputGradient: BNNSNDArrayDescriptor, generatingInputGradient inputGradient: BNNSNDArrayDescriptor, generatingBetaGradient betaGradient: BNNSNDArrayDescriptor? = nil, generatingGammaGradient gammaGradient: BNNSNDArrayDescriptor? = nil) throws

Parameters

  • batchSize:

    The number of input-output pairs.

  • input:

    The descriptor of the input.

  • output:

    The descriptor of the output.

  • outputGradient:

    The descriptor of the output gradient.

  • inputGradient:

    The descriptor of the input gradient.

  • betaGradient:

    The descriptor of the beta gradient.

  • gammaGradient:

    The descriptor of the gamma gradient.

See Also

Applying a Normalization Layer