Difference between torch.flatten() and nn.Flatten()

前端 未结 0 637
北恋
北恋 2021-02-01 20:34

I am a beginner with PyTorch and I was just wondering what is difference between torch.flatten and torch.nn.Flatten()?

Thanks :)

相关标签:
回答
  • 消灭零回复
提交回复
热议问题