小编Som*_*one的帖子

从 pytorch 张量中按索引删除行

我有一个大小为 的 pytorch 张量torch.Size([4, 3, 2])

tensor([[[0.4003, 0.2742],
     [0.9414, 0.1222],
     [0.9624, 0.3063]],

    [[0.9600, 0.5381],
     [0.5758, 0.8458],
     [0.6342, 0.5872]],

    [[0.5891, 0.9453],
     [0.8859, 0.6552],
     [0.5120, 0.5384]],

    [[0.3017, 0.9407],
     [0.4887, 0.8097],
     [0.9454, 0.6027]]])
Run Code Online (Sandbox Code Playgroud)

我想删除第二行,使张量变为torch.Size([3, 3, 2])

tensor([[[0.4003, 0.2742],
     [0.9414, 0.1222],
     [0.9624, 0.3063]],

    [[0.5891, 0.9453],
     [0.8859, 0.6552],
     [0.5120, 0.5384]],

    [[0.3017, 0.9407],
     [0.4887, 0.8097],
     [0.9454, 0.6027]]])
Run Code Online (Sandbox Code Playgroud)

如何删除 3D 张量的第 n 行?

python pytorch

0
推荐指数
1
解决办法
7987
查看次数

标签 统计

python ×1

pytorch ×1