[人工智能] Pytorch--Tensor的索引与切片

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 人工智能 -> Pytorch--Tensor的索引与切片 -> 正文阅读

[人工智能]Pytorch--Tensor的索引与切片

Pytorch--Tensor的索引与切片

1、直接用索引的方式

在这里插入图片描述

相当于创建一个CNN的Tensor
a[0].shape
直接指向对应维度的第几个

2、：号的方式

在这里插入图片描述
三种情况

start：#指从start包括start到最后
：end #指从开始到end不包括end
start：end#指从start到end，包括start不包括end

特殊情况

start：end：step #分别代表开始位置、结束位置、步长

在这里插入图片描述

3、index_select

a.index_select(dim,tensor)

指定某一维度的tensor
在这里插入图片描述

4、…

在这里插入图片描述
指全部维度

5、 .masked_select()

在这里插入图片描述
代码解释
先通过randn（3，4）创建一个dim为2size为【3，4】的tensor
再通过.ge(0.5)使大于0.5的数字全部置为1，小于0.5的置为0
在通过.maskend_select(x,mask)找出x中原来大于0.5的数字即0.5404，0.6040，1.5771