03 2021 档案

动手实现卷积
摘要：手写一个卷积操作 主要目的在于熟悉torch中的indexing操作。 首先，要介绍torch中的unfold操作。首先这个函数的输入必须是一个4-dimensional的tensor。还要设置 一些常用的参数，例如kernel_size， stride等等。然后就会被resize成为一个(B, C 阅读全文
posted @ 2021-03-17 10:41 John_Ran 阅读(255) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示