在Tensorflow占位符内还原值

时间:2018-12-06 23:26:59

标签: tensorflow

我有一个占位符,其中包含零和一(Int32类型)。在该占位符中,如何将所有1反转为zer,将所有0反转为1?

A=[[0,1,1],
 [1,0,0]] 
 type=int32

C=[[1,0,0],
 [0,1,1]]
 type=int32

1 个答案:

答案 0 :(得分:2)

以您的示例作为(非常非常简单)解决方案的基础:

# numpy version:
import numpy as np
A = np.array([[0,1,1],
             [1,0,0]], dtype=np.uint32)     
C = 1 - A
print(C)
>> [[1 0 0]
   [0 1 1]]

# tensorflow version:
import tensorflow as tf
A = tf.constant([[0,1,1],
                [1,0,0]], dtype=tf.int32)
C = 1 - A
with tf.Session() as sess:
    print(sess.run(C))
>> [[1 0 0]
   [0 1 1]]

希望有帮助。