@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

4 lines (3 loc) • 241 B

TypeScript

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