torchmd-net
torchmd-net copied to clipboard
Added vector output from skew-symmetric tensor
Allowing vector output from TensorNet. This will allow the use of EquivariantScalar
or EquivariantVectorOutput
from TensorNet
.
Resolves #297.