我正在努力寻找这个难题的解决方案 . 我有：
mask ：形状为2d的张量=（batch_size，n）由1和0组成
n_neg ：类型为int的1d张量=（batch_size，）
我想要做的是随机地从0到1更改每个掩码行中的元素，如n_neg的对应行所示 . 通过先前的构造，这将始终是可能的，即 mask 的第i行中的零的数量总是大于 n_neg [i] . 因为它只是利用tensorflow操作来做它 .
有什么想法怎么做？提前致谢