Cambridge-ICCS / FTorch

A library for directly calling PyTorch ML models from Fortran.
https://cambridge-iccs.github.io/FTorch/
MIT License
48 stars 11 forks source link

Overload elementary operations for `torch_tensor` #138

Open jwallwork23 opened 1 week ago

jwallwork23 commented 1 week ago

Related to #111.

Recently we've discussed implementing automatic differentiation (AD) in FTorch using the underlying autograd AD tool built into Torch. The first step to achieve this would be to overload elementary operations for the Fortran torch_tensor class: assignment, addition, multiplication of tensors, multiplication of tensor by scalar, etc.