pytorch删除行全为0的元素

啥都亿点点的研究生

于 2021-11-23 17:34:58 发布

阅读量3.5k

点赞数 1

文章标签：计算机视觉 opencv python

本文链接：https://blog.youkuaiyun.com/lichaoqi1/article/details/121498578

版权

def del_tensor_0_cloumn(Cs):

    idx = torch.all(Cs[..., :] == 0, axis=1)

    index=[]
    for i in range(idx.shape[0]):
        if not idx[i].item():
            index.append(i)


    index=torch.tensor(index)

    Cs = torch.index_select(Cs, 0, index)
    return Cs