Pytorch Tensor的索引与切片例子

作者:洪流之源 时间:2022-12-08 16:28:10 

1. Pytorch风格的索引

根据Tensor的shape,从前往后索引,依次在每个维度上做索引。

示例代码:


import torch

a = torch.rand(4, 3, 28, 28)
print(a[0].shape) #取到第一个维度
print(a[0, 0].shape) # 取到二个维度
print(a[1, 2, 2, 4]) # 具体到某个元素

上述代码创建了一个shape=[4, 3, 28, 28]的Tensor,我们可以理解为4张图片,每张图片有3个通道,每个通道是28x28的图像数据。a代表这个Tensor,a后面跟着的列表[]表示对Tensor进行索引,a的维度dim = 4,决定了[]中的元素个数不能超过4个,[]中的值表示对应维度上的哪一个元素,比如 a[0]表示取第一个维度上的第一个元素,可以理解为第一张图片,a[1]表示取第一个维度上的第二个元素,可以理解为第二张图片。a[0, 0]表示取第一个维度上第一个元素的与第二个维度上的第一个元素,也就是第一张图片第一个通道的元素。a[1, 2, 2, 4]表示取第第一个维度上的第二个元素与第二个维度上的第三个元素与第三个维度上的第三个元素与第四个维度上的第5个元素,也就是第二张图片第三个通道第三行第四列的像素值是一个标量值。

输出结果:


torch.Size([3, 28, 28])
torch.Size([28, 28])
tensor(0.1076)

2. python风格的索引

示例代码:


import torch

# 譬如:4张图片,每张三个通道,每个通道28行28列的像素
a = torch.rand(4, 3, 28, 28)

# 在第一个维度上取后0和1,等同于取第一、第二张图片
print(a[:2].shape)

# 在第一个维度上取0和1,在第二个维度上取0,
# 等同于取第一、第二张图片中的第一个通道
print(a[:2, :1, :, :].shape)

# 在第一个维度上取0和1,在第二个维度上取1,2,
# 等同于取第一、第二张图片中的第二个通道与第三个通道
print(a[:2, 1:, :, :].shape)

# 在第一个维度上取0和1,在第二个维度上取1,2,
# 等同于取第一、第二张图片中的第二个通道与第三个通道
print(a[:2, -2:, :, :].shape)

# 使用step隔行采样
# 在第一、第二维度取所有元素,在第三、第四维度隔行采样
# 等同于所有图片所有通道的行列每个一行或者一列采样
# 注意:下面的代码不包括28
print(a[:, :, 0:28:2, 0:28:2].shape)
print(a[:, :, ::2, ::2].shape) # 等同于上面语句

注意:负值的索引即表示倒数第几个元素,-2就是倒数第二个元素。

输出结果:


torch.Size([2, 3, 28, 28])
torch.Size([2, 1, 28, 28])
torch.Size([2, 2, 28, 28])
torch.Size([2, 2, 28, 28])

3. index_select()选择特定索引

选择特定下标有时候很有用,比如上面的a这个Tensor可以看作4张RGB(3通道)的MNIST图像,长宽都是28px。那么在第一维度上可以选择特定的图片,在第二维度上选择特定的通道,在第三维度上选择特定的行等:


# 选择第一张和第三张图
print(a.index_select(0, torch.tensor([0, 2])).shape)

# 选择R通道和B通道
print(a.index_select(1, torch.tensor([0, 2])).shape)

# 选择图像的0~8行
print(a.index_select(2, torch.arange(8)).shape)

注意:index_select()的第二个索引参数必须是Tensor类型

输出结果:


torch.Size([2, 3, 28, 28])
torch.Size([4, 2, 28, 28])
torch.Size([4, 3, 8, 28])

4. 使用 ... 索引任意多的维度


import torch

a = torch.rand(4, 3, 28, 28)

# 等与a
print(a[...].shape)

# 第一张图片的所有维度
print(a[0, ...].shape)

# 所有图片第二通道的所有维度
print(a[:, 1, ...].shape)

# 所有图像所有通道所有行的第一、第二列
print(a[..., :2].shape)

输出结果:


torch.Size([4, 3, 28, 28])
torch.Size([3, 28, 28])
torch.Size([4, 28, 28])
torch.Size([4, 3, 28, 2])

5. 使用mask索引

示例代码:


import torch

a = torch.randn(3, 4)
print(a)

# 生成a这个Tensor中大于0.5的元素的掩码
mask = a.ge(0.5)
print(mask)

# 取出a这个Tensor中大于0.5的元素
val = torch.masked_select(a, mask)
print(val)
print(val.shape)

输出结果:


tensor([[ 0.2055, -0.7070, 1.1201, 1.3325],
   [-1.6459, 0.9635, -0.2741, 0.0765],
   [ 0.2943, 0.1206, 1.6662, 1.5721]])
tensor([[0, 0, 1, 1],
   [0, 1, 0, 0],
   [0, 0, 1, 1]], dtype=torch.uint8)
tensor([1.1201, 1.3325, 0.9635, 1.6662, 1.5721])
torch.Size([5])

注意:最后取出的 大于0.5的Tensor的shape已经被打平。

6. take索引

take索引是在原来Tensor的shape基础上打平,然后在打平后的Tensor上进行索引。

示例代码:


import torch

a = torch.tensor([[3, 7, 2], [2, 8, 3]])
print(a)
print(torch.take(a, torch.tensor([0, 1, 5])))

输出结果:


tensor([[3, 7, 2],
   [2, 8, 3]])
tensor([3, 7, 3])

来源:https://blog.csdn.net/weicao1990/article/details/93599947

标签:Pytorch,Tensor,索引,切片
0
投稿

猜你喜欢

  • python中星号变量的几种特殊用法

    2021-03-20 10:57:50
  • Python实现实时显示进度条的六种方法

    2022-03-27 02:09:35
  • 微信小程序实现上传图片功能

    2024-05-02 17:29:03
  • Python 3.x 新特性及10大变化

    2023-02-05 09:17:36
  • python的多线程原来可以这样解

    2021-02-22 16:12:18
  • insert select与select into 的用法使用说明

    2012-01-05 18:47:58
  • 选择一个优秀正文字体的15个技巧

    2008-03-20 13:36:00
  • Div的浮动+循环(描述的不清楚,请看图)

    2008-09-22 20:21:00
  • python基于property()函数定义属性

    2022-11-01 04:55:14
  • 将Reporting services的RDL文件拷贝到另外一台机器时报Data at the root level is invalid的解决方法

    2012-07-11 15:33:45
  • Python实现把json格式转换成文本或sql文件

    2022-06-03 14:52:23
  • Python下rrdtool模块的基本使用方法

    2022-08-12 13:08:28
  • Python xlwings插入Excel图片的实现方法

    2023-11-23 05:53:18
  • 在Python中通过机器学习实现人体姿势估计

    2022-05-20 13:08:25
  • python查看模块,对象的函数方法

    2021-07-11 19:45:10
  • Python+OpenCV图像处理—— 色彩空间转换

    2022-02-12 16:28:01
  • python3实现往mysql中插入datetime类型的数据

    2024-01-20 12:45:58
  • C#动态创建Access数据库及密码的方法

    2024-01-17 21:37:21
  • Python的collections模块中namedtuple结构使用示例

    2022-06-01 19:59:42
  • Python 字典dict使用介绍

    2021-09-23 16:58:51
  • asp之家 网络编程 m.aspxhome.com