Skip to main content

hierarchical_forward

Function hierarchical_forward 

Source
pub fn hierarchical_forward(
    query: Float32Array,
    layer_embeddings: Vec<Vec<Float32Array>>,
    gnn_layers_json: Vec<String>,
) -> Result<Float32Array>
Expand description

Hierarchical forward pass through GNN layers

§Arguments

  • query - The query vector (Float32Array)
  • layer_embeddings - Embeddings organized by layer (Array of Array of Float32Array)
  • gnn_layers_json - JSON array of serialized GNN layers

§Returns

Final embedding after hierarchical processing as Float32Array

§Example

const query = new Float32Array([1.0, 0.0]);
const layerEmbeddings = [[new Float32Array([1.0, 0.0]), new Float32Array([0.0, 1.0])]];
const layer1 = new RuvectorLayer(2, 2, 1, 0.0);
const layers = [layer1.toJson()];
const result = hierarchicalForward(query, layerEmbeddings, layers);