counts = multinomial.Multinomial(10, fair_probs).sample((500,)) cum_counts = counts.cumsum(dim=0)
时间: 2024-05-28 10:10:42 浏览: 164
sql中count或sum为条件的查询示例(sql查询count)
这是使用 PyTorch 中的 multinomial 函数生成 500 个样本,每个样本都是从一个大小为 10 的多项式分布中随机采样得到的。fair_probs 是一个包含每个类别出现概率的张量。cum_counts 是一个包含每个样本的累计计数的张量。在这个张量中,每一行的第 i 个元素表示前 i 个样本中第一个出现类别 j 的位置。
阅读全文