torcharrow.Column.transform

Column.transform(func: Callable, dtype: Optional[DType] = None, format: str = 'column', columns: Optional[List[str]] = None)

类似于 map()，但一次对一批行的微批次调用可调用对象。默认情况下，该列作为 TorchArrow 列传递给可调用对象。如果 format=’python’，则输入将转换为 python 类型。如果 format=’torch’，则输入将转换为 PyTorch 类型，如果结果类型 != 项目类型并且可调用对象上缺少类型提示，则需要 dtype。