pymoose.add_n#

pymoose.add_n(arrays, placement=None)[source]#

Elementwise addition of a collection of tensors.

Parameters
  • arrays – Tuple or list of tensors w/ identical shape and dtype.

  • placement – An optional Placement to pin this operation to.

Returns

A tensor containing the elementwise sum of all input tensors.