Tensor矩阵索引的用法（持续补充）

Laopiiiii

已于 2022-07-31 20:17:07 修改

阅读量334

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习暑期课程文章标签：矩阵 python 深度学习

于 2022-07-31 20:13:49 首次发布

本文链接：https://blog.csdn.net/weixin_43479991/article/details/126089863

深度学习暑期课程专栏收录该内容

4 篇文章

订阅专栏

本文介绍了使用PyTorch进行矩阵取行与列的操作，通过实例展示了如何选取指定范围的行和列。从一个四维张量出发，逐步讲解了如何获取#行<=1的所有列、#行>1的所有列、行列限制的子矩阵和特定行列区域。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

取行列

import torch

x = torch.tensor([[1,2,3],[4,5,6],[7,8,9],[0,0,0]])
y1 = x[:1,:] #行<=1,所有列
y2 = x[1:,:] #行>1,所有列,等价于x[1:]
y3 = x[:1,:2] #行<=1,列<=2
y4 = x[1:,2:] #行>1,列>2

print("原矩阵\n",x)

print("#行<=1,所有列\n",y1)
print("#行>1,所有列\n",y2)
print("#行<=1,列<=2\n",y3)
print("#行>1,列>2\n",y4)

结果

原矩阵
 tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9],
        [0, 0, 0]])
#行<=1,所有列
 tensor([[1, 2, 3]])
#行>1,所有列 tensor([[4, 5, 6],
        [7, 8, 9],
        [0, 0, 0]])
#行<=1,列<=2
 tensor([[1, 2]])
#行>1,列>2
 tensor([[6],
        [9],
        [0]])