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))