python中张量tensor的切片问题

张量切片详解
本文详细介绍了Python中张量的基本概念及其切片操作方法。通过具体示例展示了如何对不同维度的张量进行切片取值,适用于初学者快速掌握张量的使用技巧。

python中的张量类似于多维数组,所以切片取值的时候要注意他的维度

例如

x = torch.rand(5,5)
print(x)
print(x[:,3])

上面的x为一个二维张量,每一个维度的大小为5

执行结果如下

可以看到取到了x的第三列的所有值

再来一个例子

y = torch.rand(5,5,5)
print(y)
print(y[1:3,:,:])

执行结果

取到的切片值

所以对于一个张量m,m[:,:,:....],m是多少维度的就有多少个被逗号隔开的值

从第一维开始顺序往后,需要取哪一维的那一部分值,就可以在:的两边加上切片范围

评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值