我有一个火炬张量如下 -
a = tensor(
[[0.2215, 0.5859, 0.4782, 0.7411],
[0.3078, 0.3854, 0.3981, 0.5200],
[0.1363, 0.4060, 0.2030, 0.4940],
[0.1640, 0.6025, 0.2267, 0.7036],
[0.2445, 0.3032, 0.3300, 0.4253]], dtype=torch.float64)
如果每行的第一个值小于 0.2,则需要删除整行。因此我需要像这样的输出 -
tensor(
[[0.2215, 0.5859, 0.4782, 0.7411],
[0.3078, 0.3854, 0.3981, 0.5200],
[0.2445, 0.3032, 0.3300, 0.4253]], dtype=torch.float64)
我尝试循环遍历张量并将有效值附加到新的空张量,但没有成功。有什么办法可以高效的得到结果吗?