tensorflow tf.data.Dataset.reduce()用法及代碼示例

借助tf.data.Dataset.reduce()方法，我們可以獲得數據集中所有元素的簡化變換tf.data.Dataset.reduce()方法。

用法： tf.data.Dataset.reduce()

返回：Return combined single result after transformation.

注意：
這些給定的示例將演示新版本的tensorflow 2.0的用法，因此，如果要運行這些示例，請在命令提示符下運行以下命令。

pip install tensorflow==2.0.0-rc2

範例1：
在這個例子中，我們可以通過使用tf.data.Dataset.reduce()方法，我們能夠從數據集中獲得所有元素的簡化變換。

# import tensorflow 
import tensorflow as tf 
  
# using tf.data.Dataset.reduce() method 
gfg = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4, 5]) 
  
print(gfg.reduce(0, lambda x, y:x + y).numpy())

輸出：

15

範例2：

# import tensorflow 
import tensorflow as tf 
  
# using tf.data.Dataset.reduce() method 
gfg = tf.data.Dataset.from_tensor_slices([[5, 10], [3, 6]]) 
  
print(gfg.reduce(0, lambda x, y:x * y).numpy())

輸出：

[15, 60]

相關用法

注：本文由純淨天空篩選整理自Jitender_1998大神的英文原創作品 Tensorflow | tf.data.Dataset.reduce()。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。