module 'torch.nn' has no attribute 'flatten'

这是因为在 PyTorch 的最新版本中,flatten 操作不再是 torch.nn 模块的一部分。相反,它是 torch.flatten 函数的一部分。

所以您需要使用 torch.flatten 函数来进行 flatten 操作,而不是 torch.nn.flatten。

  • 3年前
  •