会员
周边
众包
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
zukang
博客园
首页
新随笔
联系
管理
订阅
2021年9月6日
PyTorch中Tensor的查找和筛选
摘要: 转自:https://blog.csdn.net/tfcy694/article/details/85332953 按照指定轴上的坐标进行过滤 index_select()沿着某tensor的一个轴dim筛选若干个坐标 >>> x = torch.randn(3, 4) # 目标矩阵 >>> x t
阅读全文
posted @ 2021-09-06 19:31 鲍曼小学生
阅读(1818)
评论(0)
推荐(0)
编辑
公告