RecordBatch 轉換為 Tensor 實例#

Arrow 提供了一種方法,可將 RecordBatch 物件轉換為具有兩個維度的 Tensor

std::shared_ptr<RecordBatch> batch;

ASSERT_OK_AND_ASSIGN(auto tensor, batch->ToTensor());
ASSERT_OK(tensor->Validate());

此轉換支援有號和無號整數類型以及浮點類型。如果 RecordBatch 具有 null 值,則當 null_to_nan 參數設定為 true 時,轉換將會成功。在這種情況下,所有類型都將升級為浮點資料類型。

std::shared_ptr<RecordBatch> batch;

ASSERT_OK_AND_ASSIGN(auto tensor, batch->ToTensor(/*null_to_nan=*/true));
ASSERT_OK(tensor->Validate());

目前僅支援行優先轉換。