Struct RecurrentOps 

pub struct RecurrentOps;

Common operations for recurrent cells リカレントセル用共通操作

Implementations

impl RecurrentOps

pub fn init_weights<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( input_size: usize, hidden_size: usize, num_gates: usize, ) -> (Variable<T>, Variable<T>)

Initialize weights using Xavier/Glorot initialization Xavier/Glorot初期化で重みを初期化

pub fn init_bias<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( hidden_size: usize, num_gates: usize, ) -> (Option<Variable<T>>, Option<Variable<T>>)

Initialize bias バイアスを初期化

pub fn linear_transform<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( input: &Variable<T>, weight: &Variable<T>, bias: Option<&Variable<T>>, ) -> Variable<T>

Linear transformation: input @ weight^T + bias 線形変換: input @ weight^T + bias

pub fn matmul_variables<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( a: &Variable<T>, b: &Variable<T>, ) -> Variable<T>

Matrix multiplication for variables Variable用の行列乗算

pub fn add_variables<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( a: &Variable<T>, b: &Variable<T>, ) -> Variable<T>

Addition for variables Variable用の加算

pub fn multiply_variables<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( a: &Variable<T>, b: &Variable<T>, ) -> Variable<T>

Multiplication for variables Variable用の乗算

pub fn subtract_from_scalar<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( var: &Variable<T>, scalar: T, ) -> Variable<T>

Subtract variable from scalar スカラーから変数を減算

pub fn transpose_variable<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( var: &Variable<T>, ) -> Variable<T>

Transpose for variables Variable用の転置

pub fn sigmoid<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( var: &Variable<T>, ) -> Variable<T>

Sigmoid activation for variables Variable用のシグモイド活性化

pub fn tanh<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( var: &Variable<T>, ) -> Variable<T>

Tanh activation for variables Variable用のtanh活性化

pub fn slice_gates<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( gates: &Variable<T>, gate_idx: usize, hidden_size: usize, ) -> Variable<T>

Slice gates from concatenated tensor 連結されたテンソルからゲートをスライス

pub fn zero_hidden_state<T: Float + Send + Sync + Debug + 'static + ScalarOperand + FromPrimitive>( batch_size: usize, hidden_size: usize, ) -> Variable<T>

Create zero hidden state ゼロ隠れ状態を作成