Python – tensorflow.cond()

TensorFlow是谷歌设计的开源Python库，用于开发机器学习模型和深度学习神经网络。

cond()如果谓词pred为真，则返回true_fn()，否则就返回false_fn()。

语法: tensorflow.cond( pred, true_fn, false_fn, name )

参数:

• pred:它是一个标量，决定了要返回的可调用对象。
• true_fn（可选）:当pred为真时，它被返回。
• false_fn（可选）:当pred为false时，它被返回。
• name（可选）:它定义了该操作的名称。

返回：它返回由可调用程序评估的结果。

示例 1:

# Importing the library
import tensorflow as tf
  
# Initializing the input
x = 5
y = 10
  
  
# Printing the input
print('x: ', x)
print('y: ', y)
  
# Calculating result
res = tf.cond(x < y, lambda: tf.add(x, y), lambda: tf.square(y))
  
# Printing the result
print('Result: ', res)

输出:

x:  5
y:  10
Result:  tf.Tensor(15, shape=(), dtype=int32)

示例 2:

# Importing the library
import tensorflow as tf
  
# Initializing the input
x = 5
y = 10
  
  
# Printing the input
print('x: ', x)
print('y: ', y)
  
# Calculating result
res = tf.cond(x > y, lambda: tf.add(x, y), lambda: tf.square(y))
  
# Printing the result
print('Result: ', res)

输出:

x:  5
y:  10
Result:  tf.Tensor(100, shape=(), dtype=int32)