torch 代码片段：从3d-tensor中选择满足条件的行数据并重新组合

从3d-tensor中选择满足条件的行数据, 重新组合成原始维度的3d-tensor, 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