helion.language.join

helion.language.join(tensor0, tensor1)

Join two tensors along a new minor dimension.

Parameters:

• tensor0 (Tensor) – First tensor to join.

• tensor1 (Tensor) – Second tensor to join. Must be broadcast-compatible with tensor0.

Returns:

A tensor with shape broadcast_shape + (2,) where broadcast_shape is the broadcast of the input shapes.

Return type:

torch.Tensor

See also

• split()