Python tensorflow.cond()用法及代码示例

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

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

用法：tensorflow.cond( pred, true_fn, false_fn, name )

参数：

pred:它是一个标量，确定可调用的返回
true_fn(可选)：当pred为true时返回。
false_fn(可选)：如果pred为假，则返回该值。
name(optional):它定义了操作的名称。

返回：它返回由callable评估的结果。

范例1：

Python3

# 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：

Python3

# 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)

相关用法

注：本文由纯净天空筛选整理自aman neekhara大神的英文原创作品 Python – tensorflow.cond()。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。