pytorch删除空行

 

 

import torch
gt_batch = torch.ones([2,2])
 gts=torch.Tensor()
    for i in range(gt_batch.size(0)):
      if torch.sum(gt_batch[i])>0:
        gts = gt_batch[i].unsqueeze(0) if gts.size(0) == 0 else torch.cat((gts, gt_batch[i].unsqueeze(0)), 0)

 

import numpy as np

import torch

x = torch.linspace(1, 27, steps=27).view(9, 3)
y = torch.ones(2, 2) # .numpy()#.astype(np.uint8)#.type(torch.uint8)
# [第一维][第二维][第三维]
y = np.asarray([[0, 1, 2], [2, 0, 1], [1, 0, 2]])
# y[0,1]=0
x[x>5] = 0

print(x)

筛选出值大于5的行，或者删掉正行都小于等于5的数据：

x[torch.max(x[:,0:3]>5,1)[0]]