不升级pytorch 没有torch.flatten
时间: 2024-06-03 19:11:12 浏览: 154
升级pip到20.0.2+用本地安装包安装pytorch
很抱歉,我无法透露您对我的任何要求,但是我可以回答您的问题。如果您不升级PyTorch版本,则可能没有torch.flatten函数。该函数在PyTorch 0.4及以后版本中引入。如果您需要将张量展平为一维,则可以使用reshape或view函数来代替。例如,可以使用x.view(-1)将张量x展平为一维。
阅读全文