Cambridge-ICCS / FTorch

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

Handle multiple outputs from models. #136

Closed jatkinson1000 closed 3 months ago

jatkinson1000 commented 3 months ago

Currently work in progress.

Requested by user in #135 Closes #3

Definitely possible, but my C++-fu is not as strong as Fortran so currently stuck in pointer hell.

Other thoughts:

jwallwork23 commented 3 months ago

Okay @jatkinson1000 @TomMelt @ma595 this is now working! Ready for review.

jwallwork23 commented 3 months ago

Updated docs - ready for re-review.

jatkinson1000 commented 3 months ago

@TomMelt It seems like this was merged somewhat prematurely. As a result I am having to update the README over in #143 to cover some of the changes made here which is not ideal.