Tensor的张量切片

最新推荐文章于 2024-03-12 20:39:22 发布

蛊惑one

最新推荐文章于 2024-03-12 20:39:22 发布

阅读量550

点赞数

分类专栏： Pytorch日积月累

本文链接：https://blog.csdn.net/guhuoone/article/details/127414759

版权

python 深度学习人工智能

Pytorch日积月累专栏收录该内容

4 篇文章 0 订阅

订阅专栏

如果需要获取张量中每一行的数据，可以使用切片来实现。

语法：tensor[start:end,number]，start和end表示张量的起始位置和结束位置，number表示要取哪一行。

例：

a = torch.randn(3, 3, dtype=torch.float)
    print(a)
    a_1 = a[:, 1]
    a_2 = a[:, 2]
    a_0 = a[:, 0]
    print("a_1:", a_1)
    print("a_2:", a_2)
    print("a_0:", a_0)

结果如下所示：

tensor([[ 0.8797,  0.9092,  0.3610],
        [ 0.7417,  0.2699, -0.6761],
        [ 0.5361,  1.2840, -0.2850]])
a_1: tensor([0.9092, 0.2699, 1.2840])
a_2: tensor([ 0.3610, -0.6761, -0.2850])
a_0: tensor([0.8797, 0.7417, 0.5361])