Unverified Commit e4994ad5 authored by Gao, Xiang's avatar Gao, Xiang Committed by GitHub
Browse files

Clean wrong documents (#175)

parent 220982fb
......@@ -202,11 +202,6 @@ class AEVComputer(torch.nn.Module):
vec = self._combinations(vec, -2)
angular_terms = self._angular_subaev_terms(*vec)
# Returned tensors has shape:
# (conformations, atoms, neighbors, ``self.radial_sublength()``)
# (conformations, atoms, pairs, ``self.angular_sublength()``)
# (conformations, atoms, neighbors)
# (conformations, atoms, pairs)
return radial_terms, angular_terms, species_
def _combinations(self, tensor, dim=0):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment