aesara.tensor.join

aesara.tensor.join(axis, *tensors_list)

Convenience function to concatenate TensorTypes along the given axis.

This function will not add the op in the graph when it is not useful. For example, in the case that the list of tensors to be concatenated is one, it will just return the tensor.

Parameters:

• axis (int (symbolic or literal)) – On which dimension should the tensors be joined? The axis must be a valid index into the shape of the tensors to be concatenated. The axis parameter may either be an integer or an object that can be converted to a scalar using as_scalar`(`axis). In the former case, the axis is fixed at construction, while in the latter it may vary over time depending on the value of the axis variable.

• tensors_list (list of TensorVariable (or list-like)) – A list of tensors to be concatenated along the given axis. The shapes of the tensors to be concatenated must be all identical, except in the dimension (axis) on which they are to be joined.