我有一个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)
我想:
我可以想象这样做:
zero_list = []
for j in range(A.size()[1]):
if torch.sum(A[:,j]) == 0:
zero_list = zero_list.append(j)
标识其元素只有0的列 但是我不知道如何从原来的张量中删除这些填充了0的列。你知道吗
如何根据索引号从pytorch张量中删除带零的列?你知道吗
谢谢你
索引要保留的列比索引要删除的列更有意义。你知道吗
或者更神秘一点
相关问题 更多 >
编程相关推荐