search

DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
https://torchdyn.org
Apache License 2.0
1.33k stars 124 forks source link

Changed boolean statement to work with arrays #211

Open Ubiquinone-dot opened 3 months ago

Ubiquinone-dot commented 3 months ago

Fixes the following error which I ran into: ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

The k1 array passed in this case is of shape (batch_size, n_atoms, 3)