UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

5 lines 349 B
import { WASMTensor } from '../../../../tensor/wasm/tensor'; export function sumSquareSparseWASM(tensor, axes, keepDims) { return new WASMTensor(tensor.values.wasmTensor.sum_square_sparse(new Uint32Array(tensor.getShape()), tensor.indices.wasmTensor, new Uint32Array(axes), keepDims), undefined, tensor.dtype); } //# sourceMappingURL=wasm.js.map