def normalize_to_0_1(volume):
    """
        Normalize the image to (0,1) to be viewed as a greyscale
    Args:
        volume: 4D tensor (B, C, H, W)

    Returns:
        Normalized volume
    """
    max_val = volume.max()
    min_val = volume.min()
    return (volume - min_val) / (max_val - min_val)