简单描述一下场景:对于一个二维的整型张量,假设每一行是一堆独立的数,但是对于每一行的数,都有一个设定好的最小值的。...我们需要做的是,对于每一行,找到第一次小于最小值的位置,并将该位置起直到行末部分的数字替换为0。是不是有点抽象?...tensorflow不能对张量进行直接赋值操作,如果你尝试修改一个tensor中的内容,会报下面的错误:
TypeError: 'Tensor' object does not support item...feed_dict = {
choose:[[5,4,3,0,1],[2,3,0,4,2],[2,3,5,4,2]],
minValue:[[3],[2],[2]]}
得到每行第一个小于最小值的位置的索引...这里,我们首先判断每个位置的数是否小于最小值,如果小于最小值,返回1,大于等于最小值,返回0,那么使用arg_max函数就可以返回第一个小于最小值的位置的索引:
x = tf.tile(tf.reshape