torch.TopkOptions
export interface TopkOptions {
/** Dimension to find top k elements along (default: -1) */
dim?: number;
/** Whether to return the largest or smallest elements (default: true) */
largest?: boolean;
/** Whether to return the elements in sorted order (default: true) */
sorted?: boolean;
}dim(number)optional- – Dimension to find top k elements along (default: -1)
largest(boolean)optional- – Whether to return the largest or smallest elements (default: true)
sorted(boolean)optional- – Whether to return the elements in sorted order (default: true)
Options for topk operations