applyInTopK(k:input:testIndices:output:axis:batchSize:filterParameters:)
Applies an in-top-k filter directly to an input.
Declaration
static func applyInTopK(k: Int, input: BNNSNDArrayDescriptor, testIndices: BNNSNDArrayDescriptor, output: BNNSNDArrayDescriptor, axis: Int, batchSize: Int, filterParameters: BNNSFilterParameters? = nil) throwsParameters
- k:
The number of entries the operation finds.
- input:
The descriptor of the input.
- testIndices:
The descriptor of the test indices.
- output:
The descriptor of the output.
- axis:
The axis along which the operation finds top-k entries.
- batchSize:
The number of input-output pairs to process.
- filterParameters:
The filter runtime parameters.