我目前正在执行以下操作:
repeat_vals = [x.shape[0] // pfinal.shape[0]] + [-1] * (len(pfinal.shape) - 1)
x = torch.cat((x, pfinal.expand(*repeat_vals)), dim=-1)
x的形状是[91,6],final的形状是[6,6],但我得到以下错误:
RuntimeError: The expanded size of the tensor (15) must match the existing size (6) at non-singleton dimension 0. Target sizes: [15, -1]. Tensor sizes: [6, 6]
1条答案
按热度按时间bqf10yzr1#
您不能扩展非单一值。此外,您不能强制
len(x)
为len(pfinal)
的倍数,因此,根据您的需要,您可以增加所需的数字,然后将超出的部分切掉。您可以修改以下内容以满足您的需要: