我有一个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张量中删除零列?
谢谢,
索引要保留的列而不是要删除的列更有意义。
valid_cols = []
for col_idx in range(A.size(1)):
if not torch.all(A[:, col_idx] == 0):
valid_cols.append(col_idx)
A = A[:, valid_cols]
或更隐秘地
valid_cols = [col_idx for col_idx, col in enumerate(torch.split(A, 1, dim=1)) if not torch.all(col == 0)]
A = A[:, valid_cols]
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句