def torch_to_ait_data(tensor: TorchTensor) -> AITData:
    """
    Convert a torch Tensor to a AITData.
    """
    return AITData(
        tensor.data_ptr(), list(tensor.size()), torch_dtype_to_string(tensor.dtype)
    )
Convert a torch Tensor to a AITData.
python/aitemplate/compiler/model.py