Contents

topK(_:kTensor:name:)

Creates a TopK operation and returns the result tensor.

Declaration

func topK(_ source: MPSGraphTensor, kTensor: MPSGraphTensor, name: String?) -> [MPSGraphTensor]

Parameters

  • source:

    Tensor containing source data.

  • kTensor:

    Tensor of the number of largest values to return.

  • name:

    The name for the operation.

Return Value

A valid MPSGraphTensor array of size 2.

Discussion

Finds the k largest values along the minor dimension of the input. The source must have at least k elements along its minor dimension. The first element of the result array corresponds to the top values, and the second element of the result array corresponds to the indices of the top values.