compositional_graph_models/models.py
def _compute_output(self, inputs: torch.Tensor) -> torch.Tensor:
        """
        Compute the outputs for this model.

        Parameters
        ----------
        inputs
            Inputs to each root module in the batch.
            Has shape [batch_size, root_function_arity, self.hparams.d_model].

        Returns
        -------
        torch.Tensor
            Whatever output this model produces.
        """
        raise NotImplementedError

        Compute the outputs for this model.

        Parameters
        ----------
        inputs
            Inputs to each root module in the batch.
            Has shape [batch_size, root_function_arity, self.hparams.d_model].

        Returns
        -------
        torch.Tensor
            Whatever output this model produces.
        
Compute the outputs for this model.
Parameters

inputs
Inputs to each root module in the batch.

Returns

torch.Tensor
Whatever output this model produces.
Compute the outputs for this model . Parameters inputs Inputs to each root module in the batch . Returns torch . Tensor Whatever output this model produces .
def _compute_output(self, inputs: torch.Tensor) -> torch.Tensor:
        raise NotImplementedError
def _compute_output ( self , inputs : torch . Tensor ) -> torch . Tensor : raise NotImplementedError
Compute the outputs for this model.
Compute the outputs for this model .
"""
        Compute the outputs for this model.

        Parameters
        ----------
        inputs
            Inputs to each root module in the batch.
            Has shape [batch_size, root_function_arity, self.hparams.d_model].

        Returns
        -------
        torch.Tensor
            Whatever output this model produces.
        """
self
inputs
torch.Tensor