From 3cbcd8266233f8ca25230ae59cd113975407bd27 Mon Sep 17 00:00:00 2001 From: Morten Hjorth-Jensen Date: Sun, 26 Nov 2023 14:37:39 +0100 Subject: [PATCH] Create simple.py --- doc/Programs/PyTorch/simple.py | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 doc/Programs/PyTorch/simple.py diff --git a/doc/Programs/PyTorch/simple.py b/doc/Programs/PyTorch/simple.py new file mode 100644 index 000000000..6850f71ce --- /dev/null +++ b/doc/Programs/PyTorch/simple.py @@ -0,0 +1,7 @@ +import torch +def compute_z(a,b,c): + r1 = torch.sub(a,b) + r2 = torch.mul(r1,2) + z = torch.add(r2,c) + return z +print(compute_z(5.0,3.0,4.0))