如何将张量中的最高K值设置为1，其余的设置为0？

时间：2018-07-19 02:20:57

标签： tensorflow tensor highest

我的张量形状是（1，500，500，1）。我只想将具有最高10,000值的单元格设置为1，其余单元格设置为0。

这是张量的示例：

dateTimeControl

如果这很麻木，则可以通过以下方式完成：

n = 500
bigtf = tf.random_uniform((1,n,n,1))

enter image description here

enter image description here

0 个答案:

没有答案
相关问题
最新问题