为了账号安全,请及时绑定邮箱和手机立即绑定

如何在pytorch中堆叠一维向量

如何在pytorch中堆叠一维向量

ibeautiful 2021-09-25 16:19:46
我试图在 pytorch 中堆叠一维张量,但堆栈函数似乎将它们解释为二维方阵。任何想法如何将一维张量堆叠成一个新的一维张量?再现性:a = torch.randn([2])b = torch.randn([3])c = torch.stack([a, b]) # want a (5,) tensorRuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 2 and 3 in dimension 1 at c:\new-builder_3\win-wheel\pytorch\aten\src\th\generic/THTensorMath.cpp:3616如果我将 a 和 b 解压缩为 (2,1) 和 (3,1) 张量或压缩它们(应该无效),错误仍然存在。这似乎也很奇怪,因为它们在未压缩时都应该是维度 1 中的大小 1(并且打印它们的大小将准确反映这一点),但错误仍然出现相同的字符对字符。此外,在维度 1 中堆叠只会使其说“...维度 0 中的 2 和 3...”。这一切都在 python 3.5.4,pytorch 0.4.1 中
查看完整描述

1 回答

  • 1 回答
  • 0 关注
  • 279 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信