帮我修改以下错误 input = torch.tensor(1,2,3,4,5,6) TypeError: tensor() takes 1 positional argument but 6 were given
时间: 2023-09-24 20:05:50 浏览: 167
Pytorch 之修改Tensor部分值方式
5星 · 资源好评率100%
当使用 `torch.tensor()` 函数时,应该将多个数值参数作为一个列表传递进去,而不是将它们分别作为参数传递。所以,你应该这样写:
```
input = torch.tensor([1, 2, 3, 4, 5, 6])
```
这将创建一个包含给定数值的一维张量。如果你需要一个多维张量,可以将列表嵌套起来,例如:
```
input = torch.tensor([[1, 2], [3, 4], [5, 6]])
```
这将创建一个形状为 `(3, 2)` 的二维张量,包含给定的数值。
阅读全文