Method cat

pointersect/inference/structures.py:1314–1339

Concatenate a list of PointersectRecord at the given dimension. It is useful to split m_shape.

(records: T.List['PointersectRecord'], dim: int)

    @staticmethod
    def cat(records: T.List['PointersectRecord'], dim: int) -> 'PointersectRecord':
        """
        Concatenate a list of PointersectRecord at the given dimension.
        It is useful to split m_shape.
        """
        out = dict()
        for name in [
            'intersection_xyz_w', 'intersection_surface_normal_w', 'intersection_rgb',
            'blending_weights', 'neighbor_point_idxs', 'neighbor_point_valid_len',
            'ray_t', 'ray_hit', 'ray_hit_logit', 'model_attn_weights',
            'refined_ray_hit',
            'intersection_plane_normals_w',
            'geometry_weights',
            'valid_neighbor_idx_mask',
            'valid_plane_normal_mask',
        ]:
            arr = [getattr(r, name, None) for r in records]
            if None in arr:
                out[name] = None
            else:
                out[name] = torch.cat(arr, dim=dim)

        if len(records) > 0:
            out['model_info'] = records[0].model_info

        return PointersectRecord(**out)

    def chunk(self, chunks: int, dim: int) -> T.List['PointersectRecord']:
        """