Contents

argSort(_:axis:name:)

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

Declaration

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

Parameters

  • tensor:

    The input tensor

  • axis:

    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