我应该如何检查张量流张量中的所有数字是否全部为二进制

Question

How should I judge if all numbers in a tensorflow Tensor are either 0 or 1?

bad_mask = tf.Variable([[0.0,1.0,0.2,0.0,0.0], [0.0,5.0,0.0,2.3,0.0]])
good_mask = tf.Variable([[0.0,1.0,1.0,0.0,0.0], [0.0,1.0,0.0,1.0,0.0]])

I want to use tf.assert .

Answer 1

Like this code (tested):

import tensorflow as tf

bad_mask = tf.Variable([[0.0,1.0,0.2,0.0,0.0], [0.0,5.0,0.0,2.3,0.0]])
good_mask = tf.Variable([[0.0,1.0,1.0,0.0,0.0], [0.0,1.0,0.0,1.0,0.0]])

x = tf.Assert( tf.reduce_all( 
    tf.logical_or( tf.equal( good_mask, 0.0 ), tf.equal( good_mask, 1.0 ) )
), [ good_mask ] )

y = tf.Assert( tf.reduce_all( 
    tf.logical_or( tf.equal( bad_mask, 0.0 ), tf.equal( bad_mask, 1.0 ) )
), [ bad_mask ] )

with tf.Session() as sess:
    sess.run( tf.global_variables_initializer() )
    print( sess.run ( x ) )
    print( sess.run ( y ) )

Will output:

None

and

InvalidArgumentError: assertion failed: [[0 1 0.2]...]
[[Node: Assert_4/AssertGuard/Assert = Assert[T=[DT_FLOAT], summarize=3, _device="/job:localhost/replica:0/task:0/device:CPU:0"](Assert_4/AssertGuard/Assert/Switch, Assert_4/AssertGuard/Assert/Switch_1)]]

as desired.

Answer 2

使用tf.unstack将张量转换为列表，然后检查所有值是否为0或1。