这是因为在 PyTorch 的最新版本中,flatten 操作不再是 torch.nn 模块的一部分。相反,它是 torch.flatten 函数的一部分。
所以您需要使用 torch.flatten 函数来进行 flatten 操作,而不是 torch.nn.flatten。
狂野的白开水 · Visual Studio 2022 版本 ...· 3 月前 · |
热心肠的茶叶 · Digital Micrograph3.2 ...· 4 月前 · |
面冷心慈的绿茶 · 使用python把数组[['a']['b'] ...· 6 月前 · |
体贴的骆驼 · Three.js使用鼠标控制插件-百度经验· 1 年前 · |
腹黑的开心果 · cmip6数据处理:delta方法是每天的数 ...· 1 年前 · |
体贴的骆驼 · Three.js使用鼠标控制插件-百度经验 1 年前 |