Skip to content

Instantly share code, notes, and snippets.

@dhruvbird
Created November 12, 2021 19:26
Show Gist options
  • Save dhruvbird/4b525dcf3db74e7cf247c413224ae275 to your computer and use it in GitHub Desktop.
Save dhruvbird/4b525dcf3db74e7cf247c413224ae275 to your computer and use it in GitHub Desktop.
Run the model
m = AddtensorsModel()
res = m(torch.Tensor([1, 2, 3]), torch.Tensor([3, 4, 5]))
print(res)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment