How to evaluate TensoRF on a dense RGBA grid? about tensorf HOT 2 OPEN

def forward_based_on_ray_points(
        self,
        rays_pts,  # [n, 3] tensor
        white_bg=True,
        is_train=False,
        ndc_ray=False,
    ):
        """
        Returns only the RGB and alpha values for each point in a dense grid,
        based on only the XYZ locations of the grid cell centers.
        """
        if ndc_ray:
            # TODO - figure this out later
        else:
            # pretending like the each ray point is a ray origin
            xyz_sampled = rays_pts.reshape((
                rays_pts.shape[0], 1, rays_pts.shape[1]
            ))
            # we only take 1 sample (and we let viewsdirs = 0, so as not to move from the origin)
            z_vals = torch.ones((xyz_sampled.shape[0], 1)).to(xyz_sampled.device)
            mask_out_bbox = (
                (xyz_sampled < self.aabb[0]) | (xyz_sampled > self.aabb[1])
            ).any(dim=-1)
            ray_valid = ~mask_out_bbox
            dists = torch.cat(
                (z_vals[:, 1:] - z_vals[:, :-1], torch.zeros_like(z_vals[:, :1])),
                dim=-1,
            )
        ...
        
        # pass in None for the viewdirs - b/c none the render modules need it
        if app_mask.any():
            app_features = self.compute_appfeature(xyz_sampled[app_mask])
            valid_rgbs = self.renderModule( 
                xyz_sampled[app_mask], None, app_features
            )
            rgb[app_mask] = valid_rgbs

        ...
        # TODO: unsure about how to get depth maps, as it requires the viewdir?
        # with torch.no_grad():
        #     depth_map = torch.sum(weight * z_vals, -1)
        #     # depth_map = depth_map + (1.0 - acc_map) * rays_chunk[..., -1]
        #     depth_map = (
        #         depth_map + (1.0 - acc_map) * torch.ones_like(xyz_sampled)[..., -1]
        #     )  # just a guess

        return rgb_map, None, alpha