UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

8 lines 256 B
export function addIndexCPU(indices, axis, count) { const result = indices.copy(); for (let i = axis; i < result.size; i += indices.shape[1]) { result.set(i, result.get(i) + count); } return result; } //# sourceMappingURL=cpu.js.map