2021年10月21日
8 Torch 中 view() & size() 用法
摘要： Torch 中 view() & size() 在阅读论文源码过程中，经常会看到如下的命令： x = x.view(x.size(0), -1) # 改变 tensor 的形态 下面，本文简单介绍一下 view() 和 size() 函数的作用： view() import torch # 用法一 阅读全文
posted @ 2021-10-21 11:18 SethDeng 阅读(722) 评论(0) 推荐(0) 编辑