PyTorch教程（三）：索引与切片

迷南。 2022-10-30 09:24 236阅读 0赞

a = torch.rand(4,3,28,28) # 4张图片，3通道，高度28，宽度28
     a[0].shape # torch.Size([3, 28, 28]) 表示第0张图片的shape是3通道，高度28，宽度28
     a[0,0].shape # torch.Size([28, 28]) 表示第0张图片第0个通道的shape
     a[0,0,2,4].shape # tensor(0.5649) 表示第0张图片第0个通道第2行第4列的值是0.5649
    
    a[:2].shape # torch.Size([2, 3, 28, 28]) 表示取第0个和第1个图片的shape
    a[:2,:1,:,:].shape # torch.Size([2, 2, 28, 28]) 表示前2个图片，第1个通道上的shape
    a[:2,1:,:,:].shape # torch.Size([2, 2, 28, 28]) 表示前2个图片，第1个通道之后的shape
    a[:2,-1:,:,:].shape # torch.Size([2, 1, 28, 28]) 表示前2个图片，第最后1个通道之后的shape
    
    a[:,:,0:28:2,0:28:2].shape # torch.Size([4, 3, 14, 14]) 每隔2个数据进行采样。
    
    a.index_select(0,torch.tensor([0,2])).shape # torch.Size([2, 3, 28, 28]) 在第0个维度上进行采集第0个图片和第2个图片。
    a.index_select(1,torch.tensor([1,2])).shape # torch.Size([4, 2, 28, 28]) 在第1个维度上进行采集第2个通道和第3个通道
    a.index_select(2,torch.arange(28)).shape # torch.Size([4, 3, 28, 28]) 在第2个维度上选取所有的行
    
    a[...].shape # torch.Size([4, 3, 28, 28])表示获取所有的维度。
    a[0,...].shape # torch.Size([3, 28, 28])

发表评论取消回复

表情：

评论列表（有 0 条评论，236人围观）

还没有评论，来说两句吧...

相关阅读

相关 python的索引与切片

目录1.一维数组的索引和切片1.二维数组的索引和切片在python的学习过程，有些同学对索引和切换会感到困惑，今天我们就来弄清楚它。1.一维数组的索引和切片如下图所示，该...

谁借莪１个温暖的怀抱￠/ 2024年05月31日 04:44/ 0 赞/ 80 阅读

相关 Python字符串的索引与切片

这篇文章主要介绍了Python字符串的索引与切片，文章围绕主题展开详细的相关资料，需要的小伙伴可以参考一下。 1、字符串的索引与获取字符串的索引方式与列表的索引方式是

ゝ一纸荒年。/ 2024年03月31日 16:10/ 0 赞/ 86 阅读

相关 PyTorch教程（三）：索引与切片

a = torch.rand(4,3,28,28) 4张图片，3通道，高度28，宽度28 a[0].shape torch.Size([3, 28, 28

迷南。/ 2022年10月30日 09:24/ 0 赞/ 237 阅读

相关 Python基础教程：索引和切片

索引（下标）索引又称下标，用来表示可迭代对象中的某个元素的位置。用正整数表示的索引值，从左向右定位，从 0 开始计数，如 0，1，2 用负整数表示的索引值，

川长思鸟来/ 2022年10月09日 12:11/ 0 赞/ 216 阅读

相关 Pytorch Tensor的索引和切片介绍

1. 一般索引从前往后提供的索引，会依次在每个维度上做索引。 import torch a = torch.rand(4, 3, 28, 2

深碍√TFBOYSˉ_/ 2022年09月12日 10:46/ 0 赞/ 217 阅读

相关 [work] pytorch的切片

torch.index\_select()的示例 x = torch.randn(3, 4) print(x) indices = torc

谁践踏了优雅/ 2022年04月05日 11:29/ 0 赞/ 239 阅读

相关 [work] pytorch切片

torch 包 `torch` 包含了多维张量的数据结构以及基于其上的多种数学操作。另外，它也提供了多种工具，其中一些可以更有效地对张量和任意类型进行序列化。它有CUDA

旧城等待，/ 2022年04月05日 11:26/ 0 赞/ 233 阅读

相关 python字符串的索引与切片

字符串的索引 python中的字符串有两种方式索引： 1. 从左到右，从0开始以次增长 2. 从右到左，从-1开始依次减少如： >>>a='p

绝地灬酷狼/ 2022年02月25日 00:12/ 0 赞/ 470 阅读

相关 Pytorch Tensor的索引与切片

1. Pytorch风格的索引根据Tensor的shape，从前往后索引，依次在每个维度上做索引。示例代码： import torch a

小灰灰/ 2022年01月07日 02:37/ 0 赞/ 489 阅读

相关作业07：字符串索引与切片

字符串索引与切片 s = 'ABCDEFG' 索引从0开始依次为 0123 取出的为字符串类型 s1 = s[0]

我会带着你远行/ 2022年01月06日 19:04/ 0 赞/ 319 阅读