Simple Python code to create a very simple model

AddTensorsModel.py

import torch

class AddTensorsModel(torch.nn.Module):
  def __init__(self):
    super().__init__();
    self.t1 = torch.Tensor([0, 5, 10])

  def helper(self, x, y):
    z = self.t1
    z = z + x + y
    return z

  def forward(self, x, y):
    return self.helper(x, y)