Contents

applyBackward(batchSize:inputA:inputB:output:outputGradient:generatingInputAGradient:generatingInputBGradient:)

Applies the layer backward to generate input gradients.

Declaration

func applyBackward(batchSize: Int, inputA: BNNSNDArrayDescriptor, inputB: BNNSNDArrayDescriptor, output: BNNSNDArrayDescriptor, outputGradient: BNNSNDArrayDescriptor, generatingInputAGradient inputAGradient: BNNSNDArrayDescriptor, generatingInputBGradient inputBGradient: BNNSNDArrayDescriptor) throws

Parameters

  • batchSize:

    The number of input-output pairs.

  • inputA:

    The descriptor of the first input.

  • inputB:

    The descriptor of the second input.

  • output:

    The descriptor of the output.

  • outputGradient:

    The descriptor of the output gradient.

  • inputAGradient:

    The descriptor of the first input gradient.

  • inputBGradient:

    The descriptor of the second input gradient.

See Also

Applying a Binary Layer