UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

5 lines (4 loc) 358 B
import { CPUTensor } from '../../../tensor/cpu/tensor'; import { SparseTensor } from '../../../tensor/sparse/tensor'; import { DType } from '../../../types'; export declare function reshapeCPU<DTpe extends DType>(tensor: SparseTensor<DTpe>, values: CPUTensor<DTpe>, indices: CPUTensor<'uint32'>, shape: readonly number[], copy: boolean): SparseTensor<DTpe>;