基于pytorch定义一个数学公式

假设我们要定义一个简单的数学公式，如 y = 2x + 1，其中 x 是输入变量，y 是输出变量。可以使用 PyTorch 来定义这个公式，如下所示：

import torch

x = torch.tensor([1.0, 2.0, 3.0])  # 定义输入变量 x
y = 2 * x + 1  # 定义输出变量 y

print(y)  # 输出结果为 tensor([3., 5., 7.])

在上述代码中，我们首先使用 PyTorch 定义了输入变量 x，然后使用数学公式 y = 2x + 1 计算输出变量 y。最后，我们使用 print 函数输出结果，得到了 tensor([3., 5., 7.]) 的输出。这表示当输入变量 x 的值分别为 1.0、2.0 和 3.0 时，输出变量 y 的值分别为 3.0、5.0 和 7.0。