Struct TransformerModel 

pub struct TransformerModel<T>
where
    T: Float + 'static + Send + Sync + Debug + Default + Copy + From<f32> + ScalarOperand + FromPrimitive + ToPrimitive + Zero + One + Sum + Display,
{ /* private fields */ }

基本的な Transformer モデル Basic Transformer model

Implementations

impl<T> TransformerModel<T>

where T: Float + 'static + Send + Sync + Debug + Default + Copy + From<f32> + ScalarOperand + FromPrimitive + ToPrimitive + Zero + One + Sum + Display,

pub fn new( vocab_size: usize, d_model: usize, nhead: usize, num_encoder_layers: usize, dim_feedforward: usize, num_classes: usize, dropout_rate: f64, max_seq_length: usize, ) -> RusTorchResult<Self>

新しい Transformer モデルを作成 Create a new Transformer model

Trait Implementations

impl<T> Debug for TransformerModel<T>

where T: Float + 'static + Send + Sync + Debug + Default + Copy + From<f32> + ScalarOperand + FromPrimitive + ToPrimitive + Zero + One + Sum + Display + Debug,

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

Formats the value using the given formatter.

impl<T> Model<T> for TransformerModel<T>

where T: Float + 'static + Send + Sync + Debug + Default + Copy + From<f32> + ScalarOperand + FromPrimitive + ToPrimitive + Zero + One + Sum + Display,

fn train(&mut self)

モデルを訓練モードに設定 Set model to training mode

fn eval(&mut self)

モデルを評価モードに設定 Set model to evaluation mode

fn mode(&self) -> ModelMode

現在のモードを取得 Get current mode

fn config(&self) -> HashMap<String, String>

モデルの設定を取得 Get model configuration

fn summary(&self) -> String

モデルの概要を表示 Display model summary

impl<T> Module<T> for TransformerModel<T>

where T: Float + 'static + Send + Sync + Debug + Default + Copy + From<f32> + ScalarOperand + FromPrimitive + ToPrimitive + Zero + One + Sum + Display,

fn forward(&self, input: &Variable<T>) -> Variable<T>

モジュールの順伝搬を実行します。 Performs the forward pass of the module.

fn parameters(&self) -> Vec<Variable<T>>

モジュールのパラメータへの参照を返します。 Returns a reference to the module’s parameters.

fn as_any(&self) -> &dyn Any

ダウンキャストのための&dyn Anyとしてのモジュールへの参照を返します。 Returns a reference to the module as &dyn Any for downcasting.

fn train(&mut self)

モジュールを訓練モードに設定します。 Sets the module to training mode.

fn eval(&mut self)

モジュールを評価モードに設定します。 Sets the module to evaluation mode.