@thi.ng/tensors/identity.js

0D/1D/2D/3D/4D tensors with extensible polymorphic operations and customizable storage

import { STORAGE } from "./storage.js";
import { Tensor2 } from "./tensor.js";
const identity = (type, size, opts) => {
  const n = size * size;
  const storage = opts?.storage || STORAGE[type];
  const data = storage.alloc(n);
  for (let i = 0; i < n; i += size + 1) data[i] = 1;
  return new Tensor2(type, storage, data, [size, size], [size, 1]);
};
export {
  identity
};