小编Joh*_*ith的帖子

从 torch.Tensor 中删除一个元素

我正在尝试从张量中删除一个项目。

在下面的示例中,如何从张量中删除第三项?

tensor([[-5.1949, -6.2621, -6.2051, -5.8983, -6.3586, -6.2434, -5.8923, -6.1901,
         -6.5713, -6.2396, -6.1227, -6.4196, -3.4311, -6.8903, -6.1248, -6.3813,
         -6.0152, -6.7449, -6.0523, -6.4341, -6.8579, -6.1961, -6.5564, -6.6520,
         -5.9976, -6.3637, -5.7560, -6.7946, -5.4101, -6.1310, -3.3249, -6.4584,
         -6.2202, -6.3663, -6.9293, -6.9262]], grad_fn=<SqueezeBackward1>)
Run Code Online (Sandbox Code Playgroud)

python torch pytorch

6
推荐指数
2
解决办法
7404
查看次数

标签 统计

python ×1

pytorch ×1

torch ×1