UNPKG
@hoff97/tensor-js
Version:
latest (0.3.0)
0.3.0
0.2.0
0.1.12
0.1.11
0.1.10
0.1.9
0.1.8
0.1.7
0.1.6
0.1.5
0.1.4
0.1.2
0.1.1
0.1.0
PyTorch like deep learning inferrence library
github.com/Hoff97/tensorjs
Hoff97/tensorjs
@hoff97/tensor-js
/
dist
/
lib
/
ops
/
sparse
/
aggregate
/
sumSquare
/
cpu.js
5 lines
•
227 B
JavaScript
View Raw
1
2
3
4
5
import
{ aggregateSparseCPU }
from
'../cpu'
;
export
function
sumSquareSparseCPU
(
tensor, axes, keepDims
) {
return
aggregateSparseCPU
(tensor, axes, keepDims,
(
a, b
) =>
a + b * b,
e
=>
e * e); }
//# sourceMappingURL=cpu.js.map