Source code for torchmdo.model

import torch
from abc import ABC, abstractmethod

Tensor = torch.Tensor
as_tensor = torch.as_tensor


class Model(ABC):
    """
    Base class that must be inherited to construct a model for use in
    optimization.
    """

    @abstractmethod
    def compute(self) -> None:
        """
        Compute and set all attributes that will be used as outputs for the model
        such as objectives and constraints.
        This method can take additional keyword arguments if desired.
        """
        pass