Struct Conv1dConfig

pub struct Conv1dConfig {
    pub channels_in: usize,
    pub channels_out: usize,
    pub kernel_size: usize,
    pub stride: usize,
    pub dilation: usize,
    pub groups: usize,
    pub padding: PaddingConfig1d,
    pub bias: bool,
    pub initializer: Initializer,
}

Configuration to create a 1D convolution layer using the init function.

Fields

channels_in: usize

The number of input channels.

channels_out: usize

The number of output channels.

kernel_size: usize

The size of the kernel.

stride: usize

The stride of the convolution.

dilation: usize

Spacing between kernel elements.

groups: usize

Controls the connections between input and output channels.

padding: PaddingConfig1d

The padding configuration.

Warning

Only symmetric padding is currently supported. As such, using Same padding with an even kernel size is not supported as it will not produce the same output size.

bias: bool

If bias should be added to the output.

initializer: Initializer

The type of function used to initialize neural network parameters

Implementations

impl Conv1dConfig

pub fn new( channels_in: usize, channels_out: usize, kernel_size: usize, ) -> Conv1dConfig

Create a new instance of the config.

impl Conv1dConfig

pub fn with_stride(self, stride: usize) -> Conv1dConfig

The stride of the convolution.

pub fn with_dilation(self, dilation: usize) -> Conv1dConfig

Spacing between kernel elements.

pub fn with_groups(self, groups: usize) -> Conv1dConfig

Controls the connections between input and output channels.

pub fn with_padding(self, padding: PaddingConfig1d) -> Conv1dConfig

The padding configuration.

pub fn with_bias(self, bias: bool) -> Conv1dConfig

If bias should be added to the output.

pub fn with_initializer(self, initializer: Initializer) -> Conv1dConfig

The type of function used to initialize neural network parameters

impl Conv1dConfig

pub fn init<B>(&self, device: &<B as Backend>::Device) -> Conv1d<B>

where B: Backend,

Initialize a new conv1d module.

Trait Implementations

impl Clone for Conv1dConfig

fn clone(&self) -> Conv1dConfig

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Config for Conv1dConfig

fn save<P>(&self, file: P) -> Result<(), Error>

where P: AsRef<Path>,

Saves the configuration to a file.

fn load<P>(file: P) -> Result<Self, ConfigError>

where P: AsRef<Path>,

Loads the configuration from a file.

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for Conv1dConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Conv1dConfig

fn deserialize<D>( deserializer: D, ) -> Result<Conv1dConfig, <D as Deserializer<'de>>::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Conv1dConfig

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

Formats the value using the given formatter.

impl Serialize for Conv1dConfig

fn serialize<S>( &self, serializer: S, ) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.