def _predict(self, state_action_tensor):
    'Use provided information to make a prediction.\n\n        Args:\n            state_action_tensor: pytorch tensor with state-action values.\n\n        Returns:\n            Predicted values for observation-action tensors.\n        '
    if (self.model is None):
        self._init_model(state_action_tensor.shape[1])
    return self.model(state_action_tensor).detach().reshape((- 1))
Use provided information to make a prediction.

Args:
    state_action_tensor: pytorch tensor with state-action values.

Returns:
    Predicted values for observation-action tensors.
def _predict(self, state_action_tensor):
    'Use provided information to make a prediction.\n\n        Args:\n            state_action_tensor: pytorch tensor with state-action values.\n\n        Returns:\n            Predicted values for observation-action tensors.\n        '
    if (self.model is None):
        self._init_model(state_action_tensor.shape[1])
    return self.model(state_action_tensor).detach().reshape((- 1))