Morten Hjorth-Jensen
8 lines
151 B
Python
8 lines
151 B
Python
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))
|