当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.math.add_n用法及代码示例

逐元素添加所有输入张量。

用法

tf.math.add_n(
    inputs, name=None
)

参数

inputs tf.Tensor 或 tf.IndexedSlices 对象的列表，每个对象都具有相同的形状和类型。 tf.IndexedSlices 对象将在添加之前转换为密集张量。
name 操作的名称(可选)。

返回

与 inputs 的元素具有相同形状和类型的 tf.Tensor。

抛出

ValueError 如果inputs 的形状和数据类型不同，或者无法推断出形状。

tf.math.add_n 执行与 tf.math.accumulate_n 相同的操作。

此操作不广播其输入。如果您需要广播，请改用 tf.math.add(或 + 运算符)。

例如：

a = tf.constant([[3, 5], [4, 8]])
b = tf.constant([[1, 6], [2, 9]])
tf.math.add_n([a, b, a])
<tf.Tensor:shape=(2, 2), dtype=int32, numpy=
array([[ 7, 16],
       [10, 25]], dtype=int32)>

相关用法

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