tensorflow不能对张量进行直接赋值操作,如果你尝试修改一个tensor中的内容,会报下面的错误:
TypeError: 'Tensor' object does not support item...这里,我们首先判断每个位置的数是否小于最小值,如果小于最小值,返回1,大于等于最小值,返回0,那么使用arg_max函数就可以返回第一个小于最小值的位置的索引:
x = tf.tile(tf.reshape...(tf.arg_max(tf.cast(choose<minValue,tf.int64),1),(-1,1)),[1,5])
输出如下,第一行得到的索引是3,第二行得到的索引是2,第三行得到的索引是0...,keep_dims=True),[1,5])
对于上面的计算,如果该行所有值都大于等于最小值,结果是0,否则,结果大于0,输出如下:
[[2 2 2 2 2]
[1 1 1 1 1]
[0 0 0...True),[1,5])
index = tf.tile(tf.expand_dims(tf.range(5,dtype=tf.int64),0),[get_shape(choose)[0],1])
输出如下