Struct TensorSourceRegistry 

pub struct TensorSourceRegistry { /* private fields */ }

Registry for managing tensor sources.

Tensor sources are named collections of data (e.g., “train”, “test”) that neural predicates can index into.

Implementations

impl TensorSourceRegistry

pub fn new() -> Self

Create a new empty registry.

pub fn add_with_metadata(&mut self, name: &str, metadata: TensorMetadata)

Add a tensor source with metadata only (for testing without Python).

pub fn add(&mut self, name: &str, tensor: PyObject, metadata: TensorMetadata)

Add a tensor source with PyTorch tensor.

pub fn set_active(&mut self, name: &str) -> Result<(), TensorSourceError>

Set the active tensor source.

pub fn active_name(&self) -> Option<&str>

Get the name of the active source.

pub fn active_size(&self) -> Result<usize, TensorSourceError>

Get the size of the active source.

pub fn get_active(&self) -> Result<&PyObject, TensorSourceError>

Get the PyTorch tensor for the active source.

pub fn get_named(&self, name: &str) -> Result<&PyObject, TensorSourceError>

Get the PyTorch tensor for a specific named source (regardless of which source is active). Used by the Stage-B existential-join forward, which reads the per-event feature batch from a fixed nsr_domain source while the per-query examples source stays active.

pub fn get_metadata(&self, name: &str) -> Option<&TensorMetadata>

Get metadata for a specific source.

pub fn contains(&self, name: &str) -> bool

Check if a source exists.

pub fn check_index(&self, index: usize) -> Result<(), TensorSourceError>

Check if an index is valid for the active source.

pub fn validate_indices( &self, indices: &[usize], ) -> Result<(), TensorSourceError>

Validate multiple indices at once.

pub fn source_names(&self) -> Vec<String>

Get names of all sources.

pub fn len(&self) -> usize

Number of sources.

pub fn is_empty(&self) -> bool

Check if registry is empty.

pub fn remove(&mut self, name: &str)

Remove a source.

pub fn clear(&mut self)

Clear all sources.

pub fn iter(&self) -> impl Iterator<Item = (&str, &TensorMetadata)>

Iterate over source names and metadata.

Trait Implementations

impl Default for TensorSourceRegistry

fn default() -> Self

Returns the “default value” for a type.