deepmd.dpmodel.descriptor.make_base_descriptor

Functions

make_base_descriptor(t_tensor[, fwd_method_name])

Make the base class for the descriptor.

Module Contents

deepmd.dpmodel.descriptor.make_base_descriptor.make_base_descriptor(t_tensor, fwd_method_name: str = 'forward')

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”.