会员
周边
众包
新闻
博问
闪存
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Flix
博客园
首页
新随笔
联系
订阅
管理
2019年8月3日
【PyTorch】PyTorch 中的 dim
摘要: PyTorch 中对 tensor 的很多操作如 sum、softmax 等都可以设置 dim 参数用来指定操作在哪一维进行。PyTorch 中的 dim 类似于 numpy 中的 axis,这篇文章来总结一下 PyTorch 中的 dim 操作。 dim 与方括号的关系 创建一个矩阵 a = to
阅读全文
posted @ 2019-08-03 20:51 Flix
阅读(6498)
评论(2)
推荐(12)
编辑
公告