pymoose.concatenate#

pymoose.concatenate(arrays, axis=0, placement=None)[source]#

Concatenation of a collection of tensors along a given axis/dimension.

Note the tensors must have similar dtype and shape along all dimensions but the given one. The dimension must already exist in the input tensors; use expand_dims() on inputs to create a new dimnesion for concatenating along.

Parameters
  • arrays – Tuple or List of tensors to be concatenated.

  • axis – Optional integer representing the dimension to concatenate along. Dimension must already exist for all input tensors.