tem = torch.ones(gt_pose.shape).cuda()pred_just_pose = torch.where(pred_just_pose<1.0,pred_just_pose,tem)gt_pose = torch.where(gt_pose<1.0,gt_pose, tem)