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]]