Contents

sliceGradientTensor(_:fwdInShapeTensor:starts:ends:strides:startMask:endMask:squeezeMask:name:)

Creates a strided-slice gradient operation and returns the result tensor.

Declaration

func sliceGradientTensor(_ inputGradientTensor: MPSGraphTensor, fwdInShapeTensor: MPSGraphTensor, starts: [NSNumber], ends: [NSNumber], strides: [NSNumber], startMask: UInt32, endMask: UInt32, squeezeMask: UInt32, name: String?) -> MPSGraphTensor

Parameters

  • inputGradientTensor:

    The input gradient.

  • fwdInShapeTensor:

    The shape of the forward pass input, that is the shape of the gradient output.

  • starts:

    An array of numbers that specify the starting points for each dimension.

  • ends:

    An array of numbers that specify the ending points for each dimension.

  • strides:

    An array of numbers that specify the strides for each dimension.

  • startMask:

    A bitmask that indicates dimensions whose starts values the operation should ignore.

  • endMask:

    A bitmask that indicates dimensions whose ends values the operation should ignore.

  • squeezeMask:

    A bitmask that indicates dimensions the operation will squeeze out from the result.

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor object