Struct TensorExtractor 

pub struct TensorExtractor { /* private fields */ }

Extracts tensor data from Arrow RecordBatches.

This struct configures which columns to extract and how to convert them.

Implementations

impl TensorExtractor

pub fn new(columns: &[&str]) -> Self

Creates a new extractor for the specified columns.

pub fn from_columns(columns: Vec<String>) -> Self

Creates an extractor from owned column names.

pub fn columns(&self) -> &[String]

Returns the column names being extracted.

pub fn extract_f32(&self, batch: &RecordBatch) -> Result<TensorData<f32>>

Extracts data as f32 tensor.

Numeric columns are converted to f32. Non-numeric columns cause an error.

Errors

Returns an error if:

• A requested column doesn’t exist
• A column contains non-numeric data

pub fn extract_f64(&self, batch: &RecordBatch) -> Result<TensorData<f64>>

Extracts data as f64 tensor.

Numeric columns are converted to f64. Non-numeric columns cause an error.

Errors

Returns an error if:

• A requested column doesn’t exist
• A column contains non-numeric data

pub fn extract_i64(&self, batch: &RecordBatch) -> Result<TensorData<i64>>

Extracts data as i64 tensor.

Integer columns are converted to i64. Non-integer columns cause an error.

Errors

Returns an error if:

• A requested column doesn’t exist
• A column contains non-integer data

Trait Implementations

impl Clone for TensorExtractor

fn clone(&self) -> TensorExtractor

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TensorExtractor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.