AttributeError: module ‘torch’ has no attribute ‘vstack’
感谢 Poe Dator 的解答,这里做个记录
错误原因:似乎是PyTorch 1.8.0 之后才支持这个 torch.vstack
解决办法:所以在这之前用 torch.cat([a,b], dim=0) 代替就行
本文介绍了在使用PyTorch时遇到的AttributeError: module ‘torch’ has no attribute ‘vstack’错误的原因及解决方法。该错误源于PyTorch版本过低不支持vstack,建议使用torch.cat([a,b],dim=0)作为替代方案。
感谢 Poe Dator 的解答,这里做个记录
错误原因:似乎是PyTorch 1.8.0 之后才支持这个 torch.vstack
解决办法:所以在这之前用 torch.cat([a,b], dim=0) 代替就行

被折叠的 条评论
为什么被折叠?