將 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());
目前僅支援行優先轉換。