Struct ResNetBlock 

pub struct ResNetBlock<T>
where
    T: Float + 'static + Send + Sync + Debug + ScalarOperand + FromPrimitive,
{ /* private fields */ }

ResNet ブロック ResNet block

Implementations

impl<T> ResNetBlock<T>

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

pub fn new( in_channels: usize, out_channels: usize, stride: usize, downsample: Option<Sequential<T>>, ) -> Self

新しい ResNet ブロックを作成 Create a new ResNet block

Trait Implementations

impl<T> Debug for ResNetBlock<T>

where T: Float + 'static + Send + Sync + Debug + ScalarOperand + FromPrimitive + Debug,

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

Formats the value using the given formatter.

impl<T> Module<T> for ResNetBlock<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.