concatenate

Graph.concatenate(tensors, axis, *, name=None)

Concatenates a list of tensors along a specified dimension.

Parameters
  • tensors (list[np.ndarray or Tensor]) – The list of tensors that you want to concatenate. All of them must have the same shape in all dimensions except axis.

  • axis (int) – The dimension along which you want to concatenate the tensors.

  • name (str, optional) – The name of the node.

Returns

The concatenated tensor.

Return type

Tensor

Notes

This node only concatenates on an existing axis, it does not create new axes. If you want to stack along a new axis or concatenate scalars, add a new axis to the tensors with [None].