张量流 | tf.data.Dataset.reduce()
借助tf.data.Dataset.reduce()
方法,我们可以使用tf.data.Dataset.reduce()
方法得到数据集中所有元素的约简变换。
Syntax : tf.data.Dataset.reduce()
Return : 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]