tensorly.concatenate

concatenate(tensors, axis=0)

Concatenate tensors along an axis.

Parameters

tensors : list of tensor

The tensors to concatenate. Non-empty tensors provided must have the same shape, except along the specified axis.

axis : int, optional

The axis to concatenate on. Default is 0.

Returns

tensor