pytorch中怎么抽取一个tensor的行?x[...,0]使用介绍
- 后端
- 猿友
- 0
很多小伙伴在学习pytorch的时候会遇到x[...,0]这样的写法，但并不了解这样的写法有什么用，小编经过实验得出了这个写法的功能为pytorch抽取tensor的行。接下来就来看看x[...,0]怎么使用吧。
实验代码如下：
b=torch.Tensor([[[[10,2],[4,5],[7,8]],[[1,2],[4,5],[7,8]]]])
print(b.size())
(1, 2, 3, 2)
print(b[…,0])
tensor([[[10., 4., 7.],
[ 1., 4., 7.]]])
print(b[…,0].size())
(1, 2, 3)
print(b[…,2])
Traceback (most recent call last):
File “”, line 1, in
IndexError: index 2 is out of bounds for dimension 3 with size 2print(b[0,…])
tensor([[[10., 2.],
[ 4., 5.],
[ 7., 8.]],
[[ 1., 2.],
[ 4., 5.],
[ 7., 8.]]])
print(b[0,…].size())
(2, 3, 2)print(b[0,…,0].size())
(2, 3)
print(b[0,…,0])
tensor([[10., 4., 7.],
[ 1., 4., 7.]])[…, 0]表示抽取tensor b的第4根轴上的第一列数字组成tensor，[0, …]表示抽取tensor b的第一根轴上的第一列数字组成tensor，[0, …, 0]表示抽取b的第一根和第四根轴上的第一列数字组成tensor。
还发现一个现象
print(b[…,0:])
tensor([[[[10., 2.],
[ 4., 5.],
[ 7., 8.]],
[[ 1., 2.],
[ 4., 5.],
[ 7., 8.]]]])print(b[…,1:])
tensor([[[[2.],
[5.],
[8.]],
[[2.],
[5.],
[8.]]]])print(b[…,2:])
tensor([], size=(1, 2, 3, 0))补充：PyTorch中[..., 0]的使用案例
1. 示例1
import torch
a = torch.rand((17, 24, 8))
b = a[..., 0]
c = a[:, :, 0]
print(b == c)结果为True
2. 示例2
import torch
a = torch.rand((64, 17, 24, 8))
b = a[..., 0]
c = a[:, :, :, 0]
print(b == c)结果为True
3. 结论
可以看出[…, 0]相当于[:, :, … :, 0]
以上就是pytorch抽取tensor的行的全部内容，希望能给大家一个参考，也希望大家多多支持W3Cschool。
免责申明：本站发布的内容（图片、视频和文字）以转载和分享为主，文章观点不代表本站立场，如涉及侵权请联系站长邮箱：xbc-online@qq.com进行反馈，一经查实，将立刻删除涉嫌侵权内容。
