torsh-graph 0.1.2

Graph neural network components for ToRSh - powered by SciRS2
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! Minimal Parameter implementation for torsh-graph
//! This is a temporary replacement for torsh_nn::Parameter to avoid compilation issues

use torsh_tensor::Tensor;

/// Minimal Parameter wrapper for torsh-graph
#[derive(Clone, Debug)]
pub struct Parameter {
    tensor: Tensor,
}

impl Parameter {
    /// Create a new parameter
    pub fn new(tensor: Tensor) -> Self {
        Self { tensor }
    }

    /// Get the underlying tensor data (clone)
    pub fn clone_data(&self) -> Tensor {
        self.tensor.clone()
    }

    /// Get a reference to the tensor
    pub fn tensor(&self) -> &Tensor {
        &self.tensor
    }
}