Source code for deepmd.pt.model.descriptor.base_descriptor

# SPDX-License-Identifier: LGPL-3.0-or-later
import torch

from deepmd.dpmodel.descriptor import (
    make_base_descriptor,
)



[docs]
BaseDescriptor = make_base_descriptor(torch.Tensor, "forward")