如何Tensorflow

Question

例如上部分的神经元采用激活函数从一层，有一个张量

a=[[1,2,3,4,5], 
    [2,3,4,5,6]] 

indices =[[1, 0, 1, 0, 0], 
     [0, 1, 0, 0, 0]] 

我只想用活化上的元素（从），其索引是具有值1 （来自b）。例如，我只想在索引[0,0]，[0,2]，[1,1]的元素上使用激活函数。

谢谢！

Answer 1

您可以使用tf.where：

tf.where(tf.cast(indices, dtype=tf.bool), tf.nn.sigmoid(a), a)

对于示例：

import tensorflow as tf 

a = tf.constant([[1,2,3,4,5], [2,3,4,5,6]], dtype=tf.float32) 
indices = tf.constant([[1, 0, 1, 0, 0], [0, 1, 0, 0, 0]], 
dtype = tf.int32) 
result = tf.where(tf.cast(indices, dtype=tf.bool), tf.nn.sigmoid(a), a) 

with tf.Session() as sess: 
    print(sess.run(result)) 

此打印：

[[ 0.7310586 2.   0.95257413 4. 5. ] 
[ 2.   0.95257413 4.   5. 6 ]]