SciProgCentre/kmath

Nd4j: Provide faster implementation of `diagonalEmbedding` in `Nd4jTensorAlgebra`

Opened this issue 3 years ago · 2 comments

CommanderTvis commented 3 years ago

Postponed because of #423.

grinisrit commented 3 years ago

It is not a question of performance only, it has to be also compliant with https://deeplearning4j.org/api/latest/org/nd4j/autodiff/functions/DifferentialFunction.html
for future use.

grinisrit commented 3 years ago

To keep it there for ref:
https://deeplearning4j.org/api/latest/org/nd4j/linalg/api/ops/impl/shape/Diag.html