Struct LocalResponseNorm 

pub struct LocalResponseNorm { /* private fields */ }

Applies Local Response Normalization as described in ImageNet Classification with Deep Convolutional Neural Networks.

Y = X / (k + (alpha / size) * sum(X^2))^beta

Where the sum is computed over a sliding window of size channels.

For odd size, the window is centered on each channel position. For even size, the window uses asymmetric padding and includes the current channel plus one extra channel on the positive side.

Should be created using LocalResponseNormConfig.

Implementations

impl LocalResponseNorm

pub fn forward<B, const D: usize>(&self, input: Tensor<B, D>) -> Tensor<B, D>

where B: Backend,

Applies Local Response Normalization on the input tensor.

Shapes

• input: [N, C, D1, D2, ..., Dk] (rank >= 3)
• output: same shape as input

Panics

Panics if the input tensor rank is less than 3.

Trait Implementations

impl<B> AutodiffModule<B> for LocalResponseNorm

where B: AutodiffBackend,

type InnerModule = LocalResponseNorm

Inner module without auto-differentiation.

fn valid(&self) -> <LocalResponseNorm as AutodiffModule<B>>::InnerModule

Returns the same module, but on the inner backend without auto-differentiation.

fn from_inner( module: <LocalResponseNorm as AutodiffModule<B>>::InnerModule, ) -> LocalResponseNorm

Wraps an inner module back into an auto-diff module.

impl Clone for LocalResponseNorm

fn clone(&self) -> LocalResponseNorm

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for LocalResponseNorm

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

Formats the value using the given formatter.

impl Display for LocalResponseNorm

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

Formats the value using the given formatter.

impl<B> Module<B> for LocalResponseNorm

where B: Backend,

type Record = EmptyRecord

Type to save and load the module.

fn visit<V>(&self, _visitor: &mut V)

where V: ModuleVisitor<B>,

Visit each tensor parameter in the module with a visitor.

fn map<M>(self, _mapper: &mut M) -> LocalResponseNorm

where M: ModuleMapper<B>,

Map each tensor parameter in the module with a mapper.

fn load_record( self, _record: <LocalResponseNorm as Module<B>>::Record, ) -> LocalResponseNorm

Load the module state from a record.

fn into_record(self) -> <LocalResponseNorm as Module<B>>::Record

Convert the module into a record containing the state.

fn to_device(self, _: &<B as BackendTypes>::Device) -> LocalResponseNorm

Move the module and all of its sub-modules to the given device.

fn fork(self, _: &<B as BackendTypes>::Device) -> LocalResponseNorm

Fork the module and all of its sub-modules to the given device.

fn collect_devices( &self, devices: Vec<<B as BackendTypes>::Device>, ) -> Vec<<B as BackendTypes>::Device>

Return all the devices found in the underneath module tree added to the given vector without duplicates.

fn devices(&self) -> Vec<<B as BackendTypes>::Device>

Return all the devices found in the underneath module tree without duplicates.

fn no_grad(self) -> Self

Each tensor in the module tree will not require grad.

fn train<AB>(self) -> Self::TrainModule

where AB: AutodiffBackend<InnerBackend = B>, Self: HasAutodiffModule<AB>,

Move the module and all of its sub-modules to the autodiff backend.

fn num_params(&self) -> usize

Get the number of parameters the module has, including all of its sub-modules.

fn save_file<FR, PB>( self, file_path: PB, recorder: &FR, ) -> Result<(), RecorderError>

where FR: FileRecorder<B>, PB: Into<PathBuf>,

Save the module to a file using the provided file recorder.

fn load_file<FR, PB>( self, file_path: PB, recorder: &FR, device: &<B as BackendTypes>::Device, ) -> Result<Self, RecorderError>

where FR: FileRecorder<B>, PB: Into<PathBuf>,

Load the module from a file using the provided file recorder.

fn quantize_weights(self, quantizer: &mut Quantizer) -> Self

Quantize the weights of the module.

impl ModuleDisplay for LocalResponseNorm

fn custom_settings(&self) -> Option<DisplaySettings>

Custom display settings for the module.

fn custom_content(&self, content: Content) -> Option<Content>

Custom attributes for the module.

fn format(&self, passed_settings: DisplaySettings) -> String

Formats the module with provided display settings.

impl ModuleDisplayDefault for LocalResponseNorm

fn content(&self, content: Content) -> Option<Content>

Attributes of the module used for display purposes.

fn num_params(&self) -> usize

Gets the number of the parameters of the module.