def METHOD_NAME(self, image_idx: int) -> TensorType["image_height", "image_width", "num_channels"]:
    """Returns a 3 channel image.
    Args:
        image_idx: The image index in the dataset.
    """
    image = torch.from_numpy(self.get_numpy_image(image_idx).astype("float32") / 255.0)
    if self._dataparser_outputs.alpha_color is not None and image.shape[-1] == 4:
        assert image.shape[-1] == 4
        image = image[:, :, :3] * image[:, :, -1:] + self._dataparser_outputs.alpha_color * (1.0 - image[:, :, -1:])
    else:
        image = image[:, :, :3]
    return image