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