tensorflow - pytorch 中的 tf.function 属性

Question

我是 pytorch 的初学者，我有一些需要在网络中实现的功能。

我的问题是：有没有像tf.function这样的方法，或者我应该使用带有变量的“class(nn.Module)”？

例如，设 X 为 10x2 矩阵。在伪代码中：

a = Variable(1.0) 
b = Variable(1.0)
Y = a*X[:,0]**2 + b*X[:,1]

Answer 1

在 PyTorch 中你不需要类似的东西tf.function，你只需要使用普通的 Python 代码（因为动态图）。

如果上述内容不能回答您的问题，请提供更详细的示例（带代码）来说明您正在尝试做什么。