Pytorch，给定批次登录的样本
给定罗吉斯Pytorch，给定批次登录的样本,pytorch,Pytorch,给定罗吉斯 #每一行都是数据记录 logits=np.array（[[0.1,0.3,0.5]，[0.3,0.1,0.5]，[0.1,0.3,0.0]） 如何使用Pytorch对每行的logit的索引进行采样？当前分发API不支持此类功能 例如，我想要的是 distribution=Categorical（logits=logits） 标签=分布。样本（尺寸=1）
#每一行都是数据记录
logits=np.array（[[0.1,0.3,0.5]，[0.3,0.1,0.5]，[0.1,0.3,0.0]）
distribution=Categorical（logits=logits）
标签=分布。样本（尺寸=1）