sample_mu = torch.median(samples, dim=0)[0] sample_sigma = samples.std(dim=0)

这两行代码使用了 PyTorch 中的函数，用于计算给定数据张量的中位数和标准差。

`torch.median(samples, dim=0)` 计算数据张量 `samples` 沿着第 0 维的中位数值和中位数的索引，返回一个元组 `(values, indices)`。由于只需要中位数值，因此使用 `[0]` 取出元组中的第一个元素。

`samples.std(dim=0)` 计算数据张量 `samples` 沿着第 0 维的标准差，并返回计算结果。这个函数的参数 `dim` 指定了计算标准差的维度，可以设置为 `None` 或不指定，表示对整个张量进行计算。