search

AMReX-Codes / pyamrex

GPU-Enabled, Zero-Copy AMReX Python Bindings including AI/ML
http://pyamrex.readthedocs.io
Other
32 stars 15 forks source link

[WIP] Helper: to_torch #194

Open ax3l opened 9 months ago

ax3l commented 9 months ago

Add helper methods to generate PyTorch tensors.

Similar convenience methods as in #88 #192