@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

4 lines (3 loc) • 227 B

TypeScript

import { Tensor } from '../tensor'; import { DataType, TypedArray } from '../types'; export declare function topkImpl<T extends Tensor>(x: TypedArray, xShape: number[], xDtype: DataType, k: number, sorted: boolean): [T, T];