从3d-tensor中选择满足条件的行数据, 重新组合成原始维度的3d-tensor, 1维度产生数量变化
# 获取预测中二分类标签为1的所有点,组成临时点集
# 逐元素比较预测标签与1是否相等,相等输出同行(点)的其余信息
# 从3d-tensor中选择满足条件的行数据, 重新组合成原始维度的3d-tensor, 1维度产生数量变化
input_pc_pos = torch.zeros((8, 2048, 10))
for i in range(0,8):
for j,data in enumerate(pc_prelabel[i,:,-1]):
if data == 1 and j < 2048: input_pc_pos[i, j,:] = pc_prelabel[i, j,:]
else: continue