Struct SpatialDropoutLayer 

pub struct SpatialDropoutLayer { /* private fields */ }

Spatial dropout for convolutional layers

Implementations

impl SpatialDropoutLayer

pub fn new(dropout_rate: f32) -> Result<Self>

Create a new spatial dropout layer

Trait Implementations

impl Debug for SpatialDropoutLayer

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

Formats the value using the given formatter.

impl Display for SpatialDropoutLayer

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

Formats the value using the given formatter.

impl Layer for SpatialDropoutLayer

fn forward(&mut self, input: &Tensor, training: TrainingMode) -> Result<Tensor>

Forward pass through the layer

fn backward(&mut self, grad_output: &Tensor) -> Result<Tensor>

Backward pass through the layer

fn parameters(&self) -> Vec<&Tensor>

Get layer parameters (weights, biases, etc.)

fn parameters_mut(&mut self) -> Vec<&mut Tensor>

Get mutable layer parameters for optimization

fn gradients(&self) -> Vec<&Tensor>

Get layer gradients

fn gradients_mut(&mut self) -> Vec<&mut Tensor>

Get mutable layer gradients

fn zero_grad(&mut self)

Zero gradients

fn name(&self) -> &str

Get layer name/type

fn output_shape(&self, input_shape: &[usize]) -> Result<Vec<usize>>

Get output shape for given input shape

fn set_training(&mut self, training: bool)

Set training mode

fn training(&self) -> bool

Check if layer is in training mode

fn to_device(&mut self, _device: Device) -> Result<()>

Move layer to device

fn clone_layer(&self) -> Result<Box<dyn Layer>>

Clone layer (for model copying)

fn num_parameters(&self) -> usize

Get number of parameters