search

yhtang / FunFact

Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
Other
57 stars 4 forks source link

Parametrized Abstract Tensors #220

Closed campsd closed 2 years ago

campsd commented 2 years ago

Add functionality to support optimization of parametrized tensors.

In backward mode: tensor parameter(s) are shown to autograd backends.

In forward mode: leaf tensors are generated from parameters.