AttributeError: module ‘torch‘ has no attribute ‘vstack‘

AttributeError: module ‘torch’ has no attribute ‘vstack’

感谢 Poe Dator 的解答，这里做个记录

错误原因：似乎是PyTorch 1.8.0 之后才支持这个 torch.vstack

解决办法：所以在这之前用 torch.cat([a,b], dim=0) 代替就行