torch_frame.utils.cat

cat(lst: list[T], dim: int = 0) → T

Concatenates a list of TensorFrame or TensorData.

Parameters:

• lst (list[TensorFrame] | list[TensorData]) – A list of tensor frames to or tensor data to concatenate.

• dim (int) – Dimension to concatenate tensor data.

Returns:

Concatenated output.

Return type:

TensorFrame | TensorData