出错:TypeError: cat() received an invalid combination of arguments - got (Tensor, Tensor, dim=int),

import torch

w = torch.Tensor(6,10,1)
h = torch.Tensor(6,10,1)
#w和h是大小为6*10*1的张量

将w和h在第二维度上(最后那个1)拼接起来,使用torch.cat()操作。

new = torch.cat(w,h,dim=2)

会出现报错:

TypeError: cat() received an invalid combination of arguments - got (Tensor, Tensor, dim=int), but expected one of:
 * (tuple of Tensors tensors, int dim, *, Tensor out)
 * (tuple of Tensors tensors, name dim, *, Tensor out)

TypeError: cat() received an invalid combination of arguments - got (Tensor, Tensor, dim=int), but expected one of:
 * (tuple of Tensors tensors, int dim, *, Tensor out)
 * (tuple of Tensors tensors, name dim, *, Tensor out)

这是因为在cat操作中忘记加括号了,不能多加也不能少加括号。最后new的大小为6*10*2

#报错代码
new = torch.cat(w,h,dim=2)
#正确代码
new = torch.cat((w,h),dim=2)
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值