Pytorch函数调用

Question

我正在尝试使用 pytorch 计算三角函数，但在通过函数调用它时遇到问题，下面是我的代码：-

def func(x,y):
  return torch.exp(torch.sin(x)/x-y)

func(torch.tensor[2,3])



Error:-
TypeError - Traceback (most recent call last)
<ipython-input-16-beb818f912f5> in <module>()
----> 1 func(torch.tensor([2, 3]))

TypeError: newf() missing 1 required positional argument: 'y'

调用函数时这段代码有什么不正确的地方？

Answer 1

您需要通过解包来完成：

func(*torch.tensor[2,3])

Pytorch函数调用

Pytorch function calling

python

pytorch