(PyTorch0.4.0) AttributeError: module 'torch' has no attribute 'flatten'

1.問題:AttributeError: module 'torch' has no attribute 'flatten'   2. 解決: 參考:'tensorflow.python.layers.layers' has no attribute 'flatten' torch沒有flatten這個函數,看來確定無疑是我的PyTorch0.4.0版本太老了。 需要更新PyTorch版本。 我估
相關文章
相關標籤/搜索