tensor_input =torch.cat(tensor_input, dim=0) TypeError: cat() received an invalid combination of arguments - got (Tensor, dim=int), but expected one of: * (tuple of Tensors tensors, int dim, *, Tensor out) * (tuple of Tensors tensors, name dim, *, Tensor out
时间: 2023-12-08 12:03:13 浏览: 234
深度学习框架_PyTorch_torch.stack()函数和torch.cat()函数
这个错误提示告诉我们 `torch.cat()` 函数的参数传递有误。`torch.cat()` 函数用于将多个张量(tensor)沿着指定维度进行拼接。正确的函数调用格式为:
```
torch.cat(tensors, dim=0, out=None)
```
其中,`tensors` 是一个张量(tensor)元组或列表,`dim` 是要拼接的维度,`out` 是输出张量(tensor),可选参数。所以,你需要将所有要拼接的张量(tensor)组成一个元组或列表,再传递给 `torch.cat()` 函数。如果你想要在第0维度上拼接张量,可以按照如下方式调用函数:
```
tensor_input = torch.cat((tensor_input,), dim=0)
```
这样就能够将 `tensor_input` 这个张量拼接起来了。
阅读全文