pytorch学习笔记3 tensor索引和切片
创始人
2024-11-11 03:36:31

dim 0 占先
在这里插入图片描述
切片 (前N或者后N个)
在这里插入图片描述
:2 表示 0到2(不包含2), 1:表示 1到末尾, -1表示最后一个元素,-2表示倒数第二个
在这里插入图片描述
0:28:2 表示从0到27隔点采样
::2 表示从头到尾隔点采样
在这里插入图片描述
上图在0维上取0和2两个元素
在这里插入图片描述
在这里插入图片描述
…表示任意都维度(根据原数据shape和给定形式推断),且都是满选
a[0,…]其实=a[0,:,:,:]=a[0]
a[:,1,…]其实=a[:,1,:,:]=a[:,1]

掩码选取
在这里插入图片描述
打平选取

在这里插入图片描述

相关内容

热门资讯

裸辞做“一人公司”,我后悔了 去年这个时候,一位以色列程序员正在东南亚旅行。他顺手把一个在脑子里转了很久的想法做成了产品,一个让任...
南京建成国内首个Pre-6G试... 4月21日,2026全球6G技术与产业生态大会在南京开幕。全息互动技术展台前,一名远在北京的工作人员...
超梵求职受邀参加“2025抖音... 超梵求职受邀参加“2025抖音巨量引擎成人教育行业生态大会”,探讨分享优质内容传播,服务万千学员。 ...
摩托罗拉Razr 2026(R... IT之家 4 月 22 日消息,摩托罗拉宣布新一代 Razr 折叠手机将于 4 月 29 日在美国发...
库克卸任,特纳斯领航:苹果新纪... 苹果首席执行官蒂姆·库克将卸任,硬件工程主管约翰·特纳斯将接任,苹果公司今天宣布此事。 库克将在夏季...