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

返回张量列表的元素总和。

用法

tf.math.accumulate_n(
    inputs, shape=None, tensor_dtype=None, name=None
)

参数

inputs Tensor 对象的列表，每个对象都具有相同的形状和类型。
shape inputs 元素的预期形状(可选)。还控制此操作的输出形状，这可能会影响其他操作中的类型推断。 None 的值表示“从 inputs 中的形状推断输入形状”。
tensor_dtype inputs 的预期数据类型(可选)。 None 的值表示“从 inputs[0] 推断输入 dtype”。
name 操作的名称(可选)。

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

抛出

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

可选地，通过 shape 和 tensor_dtype 进行形状和类型检查，否则，这些是推断的。

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

例如：

a = tf.constant([[1, 2], [3, 4]])
b = tf.constant([[5, 0], [0, 6]])
tf.math.accumulate_n([a, b, a])  # [[7, 4], [6, 14]]

# Explicitly pass shape and type
tf.math.accumulate_n([a, b, a], shape=[2, 2], tensor_dtype=tf.int32)
                                                               # [[7,  4],
                                                               #  [6, 14]]

相关用法

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