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模拟登录验证码(代码简单)
Feb 06 Python
Python基于pygame模块播放MP3的方法示例
Sep 30 Python
Python实现对百度云的文件上传(实例讲解)
Oct 21 Python
python使用筛选法计算小于给定数字的所有素数
Mar 19 Python
Python3.6通过自带的urllib通过get或post方法请求url的实例
May 10 Python
python自动发送邮件脚本
Jun 20 Python
Python中如何导入类示例详解
Apr 17 Python
详解Python 中sys.stdin.readline()的用法
Sep 12 Python
Python datetime 格式化 明天,昨天实例
Mar 02 Python
python闭包、深浅拷贝、垃圾回收、with语句知识点汇总
Mar 11 Python
Python3.6 中的pyinstaller安装和使用教程
Mar 16 Python
Python使用windows设置定时执行脚本
Nov 12 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 强制性文件下载功能的函数代码(任意文件格式)
2010/05/26 PHP
PHP中的替代语法简介
2014/08/22 PHP
在Yii2中使用Pjax导致Yii2内联脚本载入失败的原因分析
2016/03/06 PHP
WordPress中设置Post Type自定义文章类型的实例教程
2016/05/10 PHP
php生成随机数/生成随机字符串的方法小结【5种方法】
2020/05/27 PHP
javascript removeChild 使用注意事项
2009/04/11 Javascript
jQuery Ajax之$.get()方法和$.post()方法
2009/10/12 Javascript
jquery的ajax请求全面了解
2013/03/20 Javascript
Knockout visible绑定使用方法
2013/11/15 Javascript
JavaScript实现简单的数字倒计时
2015/05/15 Javascript
浅谈node.js中async异步编程
2015/10/22 Javascript
使用Promise解决多层异步调用的简单学习心得
2016/05/17 Javascript
jquery插件autocomplete用法示例
2016/07/01 Javascript
使用do...while的方法输入一个月中所有的周日(实例代码)
2016/07/22 Javascript
vue如何判断dom的class
2018/04/26 Javascript
ES6关于Promise的用法详解
2018/05/07 Javascript
jquery实现的简单轮播图功能【适合新手】
2018/08/17 jQuery
Vue封装的可编辑表格插件方法
2018/08/28 Javascript
Vue中添加滚动事件设置的方法详解
2020/09/14 Javascript
Nuxt的路由配置和参数传递方式
2020/11/06 Javascript
[54:56]DOTA2上海特级锦标赛主赛事日 - 5 总决赛Liquid VS Secret第三局
2016/03/06 DOTA
[08:40]Navi Vs Newbee
2018/06/07 DOTA
基于Python3 逗号代码 和 字符图网格(详谈)
2017/06/22 Python
python出现"IndentationError: unexpected indent"错误解决办法
2017/10/15 Python
Python获取指定文件夹下的文件名的方法
2018/02/06 Python
python3使用SMTP发送简单文本邮件
2018/06/19 Python
Scrapy框架基本命令与settings.py设置
2020/02/06 Python
利用Python实现某OA系统的自动定位功能
2020/05/27 Python
英国第一摩托车和摩托车越野配件商店:GhostBikes
2019/03/10 全球购物
体育专业学生自我评价范文
2014/01/17 职场文书
3的组成教学反思
2014/04/30 职场文书
入党介绍人评语
2014/05/06 职场文书
一份没有按时交货失信于客户的检讨书
2014/09/19 职场文书
表扬稿格式范文
2015/01/16 职场文书
就业意向协议书
2015/01/29 职场文书