@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
10 lines • 568 B
JavaScript
import { WASMTensor } from '../../../../tensor/wasm/tensor';
export function reduceLogSumExpSparseWASM(tensor, axes, keepDims) {
if (tensor.dtype !== 'float32' && tensor.dtype === 'float64') {
throw new Error('Reduce log sum exp expects tensor datatype to be float, but found ' +
tensor.dtype);
}
return new WASMTensor(tensor.values
.wasmTensor.reduce_log_sum_exp_sparse(new Uint32Array(tensor.getShape()), tensor.indices.wasmTensor, new Uint32Array(axes), keepDims), undefined, tensor.dtype);
}
//# sourceMappingURL=wasm.js.map