pytorch

1
def forward(self,
2
    x: Tensor,
3
    y: Tensor) -> Tensor:
4
  return torch.add(torch.mul(x, 2), y)
5