Tensorflow | TF.data.dataset.reduce()
原文:https://www.geesforgeks.org/tensorflow-TF-data-dataset-reduce/
借助**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())
输出:
Fifteen
例 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]