var k: Int
s11TensorUtils16TopKLogitsWarperV1kSivp
2Q9LU
import TensorUtils
struct TopKLogitsWarper
Top-K. Select the k most-probable element indices from arr and return both the indices (from the original array) and their probabilities.
arr
@frozen struct Int
A signed integer value type.
init(k: Int)
func warp(indices: [Int], logits: [Float]) -> (indices: [Int], logits: [Float])