deepmd.dpmodel.atomic_model.make_base_atomic_model
Module Contents
Functions
| Make the base class for the atomic model. |
- deepmd.dpmodel.atomic_model.make_base_atomic_model.make_base_atomic_model(t_tensor, fwd_method_name: str = 'forward_atomic')[source]
Make the base class for the atomic model.
- Parameters:
- t_tensor
The type of the tensor. used in the type hint.
- fwd_method_name
Name of the forward method. For dpmodels, it should be “call”. For torch models, it should be “forward”.