adtzlr / tensortrax

Differentiable Tensors based on NumPy Arrays

Home Page:https://tensortrax.readthedocs.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Support external scalar-valued functions

adtzlr opened this issue · comments

along with their gradient and hessian.

Note: The math methods inside the function/gradient/hessian must support arbitrary numbers of elementwise-operating trailing axes.