op_eye function

Return a 2-D tensor with ones on the diagonal and zeros elsewhere.

  • Maintainer: Tomasz Kalinowski
  • License: MIT + file LICENSE
  • Last published: 2025-05-04