2018年8月7日
PyTorch查看Sequential中tensor维度的方法
摘要： 需要定义一个回调函数： 然后对需要查看的层注册钩子： 在查看完后移除钩子： 阅读全文
posted @ 2018-08-07 01:55 木易修 阅读(972) 评论(0) 推荐(0) 编辑