时间戳

首页 > 解决方案 > Pytorch 函数调用

python - Pytorch 函数调用

问题描述

我正在尝试使用 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'

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

标签： pythonpytorch

解决方案

您需要通过拆包来做到这一点：

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

推荐阅读