bottomK(_:axis:k:name:)
Creates a BottomK operation and returns the value and indices tensors.
Declaration
func bottomK(_ source: MPSGraphTensor, axis: Int, k: Int, name: String?) -> [MPSGraphTensor]Parameters
- source:
Tensor containing source data.
- axis:
The dimension along which to compute the BottomK values.
- k:
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 smallest 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 bottom values, and the second array corresponds to the indices of the bottom values.