UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

4 lines (3 loc) 223 B
import { CPUTensor } from '../../tensor/cpu/tensor'; import { DType } from '../../types'; export declare function gather<DTpe extends DType>(x: CPUTensor<DTpe>, axis: number, indices: CPUTensor<'uint32'>): CPUTensor<DTpe>;