Allow tensor of arbitrary shape to be passed to `Solution`

NeuroDiffGym / neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.

http://pypi.org/project/neurodiffeq/

MIT License

691 stars 90 forks source link

Allow tensor of arbitrary shape to be passed to `Solution` #135

Closed shuheng-liu closed 3 years ago

shuheng-liu commented 3 years ago

Somehow this is already supported in a previous version and I missed that. Just added documentation for this in 71e98be.