📌  相关文章
📜  张量流 | tf.data.Dataset.reduce()

📅  最后修改于: 2022-05-13 01:55:47.293000             🧑  作者: Mango

张量流 | tf.data.Dataset.reduce()

借助tf.data.Dataset.reduce()方法,我们可以使用tf.data.Dataset.reduce()方法得到数据集中所有元素的约简变换。

笔记 :
这些给出的示例将演示新版本 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())

输出 :

示例 #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())

输出 :