Python – tensorflow.cond()
TensorFlow is open-source Python library designed by Google to develop Machine Learning models and deep learning neural networks.
cond() return true_fn() if the predicate pred is true otherwise it returns false_fn().
Syntax: tensorflow.cond( pred, true_fn, false_fn, name )
Parameters:
- pred: It is a scalar which determines the callable to return
- true_fn(optional): It is returned when pred is true.
- false_fn(optional): It is returned when pred is false.
- name(optional): It defines the name for the operation.
Return: It returns the result evaluated by callable.
Example 1:
Python3
import tensorflow as tf
x = 5
y = 10
print ( 'x: ' , x)
print ( 'y: ' , y)
res = tf.cond(x < y, lambda : tf.add(x, y), lambda : tf.square(y))
print ( 'Result: ' , res)
|
Output:
x: 5
y: 10
Result: tf.Tensor(15, shape=(), dtype=int32)
Example 2:
Python3
import tensorflow as tf
x = 5
y = 10
print ( 'x: ' , x)
print ( 'y: ' , y)
res = tf.cond(x > y, lambda : tf.add(x, y), lambda : tf.square(y))
print ( 'Result: ' , res)
|
Output:
x: 5
y: 10
Result: tf.Tensor(100, shape=(), dtype=int32)
Last Updated :
26 Jun, 2020
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...