@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
5 lines (4 loc) • 358 B
TypeScript
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>;