We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f513b7b commit 5acb937Copy full SHA for 5acb937
1 file changed
deepmd/dpmodel/model/transform_output.py
@@ -100,7 +100,9 @@ def communicate_extended_output(
100
if vdef.r_differentiable:
101
if model_ret[kk_derv_r] is not None:
102
derv_r_ext_dims = list(vdef.shape) + [3] # noqa:RUF005
103
- mapping = xp.reshape(mapping, tuple(mldims + [1] * len(derv_r_ext_dims)))
+ mapping = xp.reshape(
104
+ mapping, tuple(mldims + [1] * len(derv_r_ext_dims))
105
+ )
106
mapping = xp.tile(mapping, [1] * len(mldims) + derv_r_ext_dims)
107
force = xp.zeros(vldims + derv_r_ext_dims, dtype=vv.dtype)
108
force = xp_scatter_sum(
0 commit comments