helion.language.split

helion.language.split(tensor)

Split the last dimension of a tensor with size two into two separate tensors.

Parameters:

tensor (Tensor) – The input tensor whose last dimension has length two.

Return type:

tuple[Tensor, Tensor]

Returns:

A tuple (lo, hi) where each tensor has the same shape as tensor without its last dimension.

See also

• join()