deepmd.dpmodel.descriptor.make_base_descriptor
Module Contents
Functions
| Make the base class for the descriptor. |
- deepmd.dpmodel.descriptor.make_base_descriptor.make_base_descriptor(t_tensor, fwd_method_name: str = 'forward')[source]
Make the base class for the descriptor.
- 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”.