【Pytorch】torch.nn.Conv2d 与 torch.nn.function.conv2d 的区别

简言之,torch.nn.Conv2d 的 weight 无法自定义,而需要手动设置 weight 时需要用到 torch.nn.function.conv2d。

参考资料:https://www.freesion.com/article/5206237781/

posted @ 2021-07-21 19:33  林東雨  阅读(240)  评论(0编辑  收藏  举报