class_docstr: Operation which trains an AlphaTensor model to learn more efficient
  matrix multiplications.
class_docstr_tok: ["Operation","which","trains","an","AlphaTensor","model","to","learn","more","efficient","matrix","multiplications","."]
docstr: Returns the trained model.
docstr_tok: ["Returns","the","trained","model","."]
code: def get_trained_model(self):
      """Returns the trained model."""
      return self._trained_model
  
file_path: nebuly/optimization/open_alpha_tensor/open_alpha_tensor/operations/training_op.py
imports: from pathlib import Path
  from typing import Tuple, Any, List
  import torch.optim
  from nebullvm.operations.base import Operation
  from open_alpha_tensor.core.modules.alpha_tensor import AlphaTensorModel
  from open_alpha_tensor.core.training import Trainer
  from open_alpha_tensor.operations.checkpoint_op import LoadCheckpointDataOp
label_desc: Use image node_id 3 for calling the TrainingOperation obj's underlying member method code with example usage: obj.get_trained_model() and returns: self