@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
5 lines (4 loc) • 318 B
TypeScript
import { CPUTensor } from '../../../../tensor/cpu/tensor';
import { SparseTensor } from '../../../../tensor/sparse/tensor';
import { DType } from '../../../../types';
export declare function reduceMeanSquareSparseCPU<DTpe extends DType>(tensor: SparseTensor<DTpe>, axes: number[], keepDims: boolean): CPUTensor<DTpe>;