tensorflow Using if condition inside the TensorFlow graph with tf.cond
Parameters
| Parameter | Details |
|---|---|
| pred | a TensorFlow tensor of type bool |
| fn1 | a callable function, with no argument |
| fn2 | a callable function, with no argument |
| name | (optional) name for the operation |
Remarks
predcannot be justTrueorFalse, it needs to be a Tensor- The function
fn1andfn2should return the same number of outputs, with the same types.
