@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

6 lines (5 loc) • 242 B

TypeScript

import { Tensor } from '../tensor'; import { Rank, TensorLike } from '../types'; declare function gatherND_(x: Tensor | TensorLike, indices: Tensor | TensorLike): Tensor<Rank>; export declare const gatherND: typeof gatherND_; export {};