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