Contents

argSort(_:axisTensor:name:)

Computes the indices that sort the elements of the input tensor along the specified axis.

Declaration

func argSort(_ tensor: MPSGraphTensor, axisTensor: MPSGraphTensor, name: String?) -> MPSGraphTensor

Parameters

  • tensor:

    The input tensor

  • axisTensor:

    The tensor dimension over which you sort the tensor

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor object with 32-bit integer data type