mirror of https://github.com/hpcaitech/ColossalAI
4 lines
194 B
Python
4 lines
194 B
Python
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"]
|
