pytorch中[..., 0]的用法说明


Posted in Python onMay 20, 2021

在看程序的时候看到了x[…, 0]的语句不是很理解,后来自己做实验略微了解,以此记录方便自己查看。

b=torch.Tensor([[[[10,2],[4,5],[7,8]],[[1,2],[4,5],[7,8]]]])
print(b.size())
(1, 2, 3, 2)
print(b[…,0])
tensor([[[10., 4., 7.],
[ 1., 4., 7.]]])
print(b[…,0].size())
(1, 2, 3)
print(b[…,2])
Traceback (most recent call last):
File “”, line 1, in
IndexError: index 2 is out of bounds for dimension 3 with size 2
print(b[0,…])
tensor([[[10., 2.],
[ 4., 5.],
[ 7., 8.]],
[[ 1., 2.],
[ 4., 5.],
[ 7., 8.]]])
print(b[0,…].size())
(2, 3, 2)
print(b[0,…,0].size())
(2, 3)
print(b[0,…,0])
tensor([[10., 4., 7.],
[ 1., 4., 7.]])

[…, 0]表示抽取tensor b的第4根轴上的第一列数字组成tensor,[0, …]表示抽取tensor b的第一根轴上的第一列数字组成tensor,[0, …, 0]表示抽取b的第一根和第四根轴上的第一列数字组成tensor。

还发现一个现象

print(b[…,0:])
tensor([[[[10., 2.],
[ 4., 5.],
[ 7., 8.]],
[[ 1., 2.],
[ 4., 5.],
[ 7., 8.]]]])
print(b[…,1:])
tensor([[[[2.],
[5.],
[8.]],
[[2.],
[5.],
[8.]]]])
print(b[…,2:])
tensor([], size=(1, 2, 3, 0))

补充:PyTorch中[..., 0]的使用案例

1. 示例1

import torch
a = torch.rand((17, 24, 8))
b = a[..., 0]
c = a[:, :, 0]
print(b == c)

结果为True

2. 示例2

import torch
a = torch.rand((64, 17, 24, 8))
b = a[..., 0]
c = a[:, :, :, 0]
print(b == c)

结果为True

3. 结论

可以看出[…, 0]相当于[:, :, … :, 0]

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python的Tornado框架实现图片上传及图片大小修改功能
Jun 30 Python
浅谈python中的占位符
Nov 09 Python
Python中Numpy包的安装与使用方法简明教程
Jul 03 Python
python调用百度语音REST API
Aug 30 Python
python实现文件的备份流程详解
Jun 18 Python
Django 查询数据库并返回页面的例子
Aug 12 Python
解决Tensorflow 内存泄露问题
Feb 05 Python
Python操作MySQL数据库的示例代码
Jul 13 Python
Python timeit模块原理及使用方法
Oct 10 Python
用python批量移动文件
Jan 14 Python
Python+Appium实现自动抢微信红包
May 21 Python
python 利用PyAutoGUI快速构建自动化操作脚本
May 31 Python
浅谈pytorch中stack和cat的及to_tensor的坑
May 20 #Python
pytorch实现手写数字图片识别
解决python3安装pandas出错的问题
May 20 #Python
python 如何在list中找Topk的数值和索引
May 20 #Python
Requests什么的通通爬不了的Python超强反爬虫方案!
python使用glob检索文件的操作
python opencv通过按键采集图片源码
You might like
php导出excel格式数据问题
2014/03/11 PHP
Yii开启片段缓存的方法
2016/03/28 PHP
iOS自定义提示弹出框实现类似UIAlertView的效果
2016/11/16 PHP
Laravel使用swoole实现websocket主动消息推送的方法介绍
2019/10/20 PHP
Javascript var变量隐式声明方法
2009/10/19 Javascript
js采用map取到id集合组并且实现点击一行选中一行
2013/12/16 Javascript
js弹窗返回值详解(window.open方式)
2014/01/11 Javascript
JavaScript语言对Unicode字符集的支持详解
2014/12/30 Javascript
jquery操作复选框checkbox的方法汇总
2015/02/05 Javascript
JavaScript实现自动消除按钮功能的方法
2015/08/05 Javascript
JS实现可展开折叠层的鼠标拖曳效果
2015/10/09 Javascript
jquery插件jquery.LightBox.js实现点击放大图片并左右点击切换效果(附demo源码下载)
2016/02/25 Javascript
用原生js做单页应用
2017/01/17 Javascript
Vue.js之slot深度复制详解
2017/03/10 Javascript
微信小程序中的swiper组件详解
2017/04/14 Javascript
微信小程序 本地图片按照屏幕尺寸处理
2017/08/04 Javascript
浅谈React高阶组件
2018/03/28 Javascript
Vue项目webpack打包部署到Tomcat刷新报404错误问题的解决方案
2018/05/15 Javascript
Angular实现模版驱动表单的自定义校验功能(密码确认为例)
2018/05/17 Javascript
Vue中 v-if 和v-else-if页面加载出现闪现的问题及解决方法
2018/10/12 Javascript
JavaScript使用闭包模仿块级作用域操作示例
2019/01/21 Javascript
微信小程序bindinput与bindsubmit的区别实例分析
2019/04/17 Javascript
小程序异步问题之多个网络请求依次执行并依次收集请求结果
2019/05/05 Javascript
Vuex实现数据共享的方法
2019/12/20 Javascript
python基础入门详解(文件输入/输出 内建类型 字典操作使用方法)
2013/12/08 Python
Python编程实现及时获取新邮件的方法示例
2017/08/10 Python
python flask框架实现传数据到js的方法分析
2019/06/11 Python
python实现提取COCO,VOC数据集中特定的类
2020/03/10 Python
Python按照list dict key进行排序过程解析
2020/04/04 Python
Selenium环境变量配置(火狐浏览器)及验证实现
2020/12/07 Python
The Hut德国站点:时装、家居用品、美容等
2016/09/23 全球购物
德国高尔夫商店:Par71.de
2020/11/29 全球购物
房地产销售计划书
2014/01/10 职场文书
企业管理毕业生求职信
2014/03/11 职场文书
2015国庆节放假通知范文
2015/07/30 职场文书
2016年七夕爱情寄语
2015/12/04 职场文书