ColossalAI/colossalai/tensor/padded_tensor/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from .api import init_as_padded_tensor, is_padded_tensor, to_padded_tensor, to_unpadded_tensor

__all__ = ["is_padded_tensor", "to_padded_tensor", "to_unpadded_tensor", "init_as_padded_tensor"]
	`from .api import init_as_padded_tensor, is_padded_tensor, to_padded_tensor, to_unpadded_tensor`

	`__all__ = ["is_padded_tensor", "to_padded_tensor", "to_unpadded_tensor", "init_as_padded_tensor"]`