如何访问Tensorflow张量的各个元素?

时间:2018-06-03 18:02:23

标签: python tensorflow keras

我正在Keras写一个模型,其中一个图层的输出给出了我想要在图像输入上应用的蒙版的坐标。问题是我找不到任何方法来执行此操作,如果您不在会话中,tf.slice()等方法不起作用。

我想做与此相同的事情:

x=dense(2)(x)
A=np.zeros((224,224, 3)) or K.zeros((224,224,3))
x=int(x)
A[x[0],x[1],:]=1

其中x来自前面的图层并在显示的图层中重新分配(来自keras的密集图层),输出是表示图片中位置的两个数字,然后我创建A,这是一个与输入相乘的掩码(输入未显示),我需要两个,使图层的输出为整数,然后使用它在位置(x [0],x [1])(和其他的,但也是其他的)添加一个到掩码A只有一个就足以知道这个方法。

0 个答案:

没有答案