def _samples_info(query_samples: Tensor) -> SamplesSummary:
    return SamplesSummary(num_chain=query_samples.size(0), num_samples=query_samples.size(1), single_sample_sz=query_samples.size()[2:])
def _samples_info(query_samples: Tensor) -> SamplesSummary:
    return SamplesSummary(num_chain=query_samples.size(0), num_samples=query_samples.size(1), single_sample_sz=query_samples.size()[2:])