ttnn.chunk

ttnn.chunk() → None

Splits a tensor into multiple chunks along a specified dimension.

:param * input_tensor: The tensor to split. :param * chunks: Number of chunks to divide the tensor into. :param * dim: Dimension along which to split.