问题背景

在知识图谱模型中需要对预测的概率进行采样选择尾节点。torch模型中使用torch.multinomial进行采样，查阅手册后替代为了与torch对应的ops.multinomial。

问题

原本在使用argmax，只选取概率最大的索引的情况下，梯度训练正常，但替换为ops.multinomial出现反向传播未定义的错误。

1.继续使用该算子有无解决方案？ 2.有无其他算子代替？