我有一个如下所示的 pytorch 张量A
:
A =
tensor([[ 4, 3, 3, ..., 0, 0, 0],
[ 13, 4, 13, ..., 0, 0, 0],
[707, 707, 4, ..., 0, 0, 0],
...,
[ 7, 7, 7, ..., 0, 0, 0],
[ 0, 0, 0, ..., 0, 0, 0],
[195, 195, 195, ..., 0, 0, 0]], dtype=torch.int32)
我想:
- 识别所有条目都等于 0 的所有列
- 仅删除所有条目都等于 0 的列
我可以想象这样做:
zero_list = []
for j in range(A.size()[1]):
if torch.sum(A[:,j]) == 0:
zero_list = zero_list.append(j)
识别其元素只有 0 的列,但我不确定如何从原始张量中删除这些填充为 0 的列。
如何根据索引号从 pytorch 张量中删除零列?
谢谢,