Struct Conv2d

pub struct Conv2d { /* private fields */ }

Implementations

impl Conv2d

pub fn new( in_channels: usize, out_channels: usize, kernel_size: usize, config: impl IntoConv2dConfig, has_bias: bool, dtype: impl Type, device: impl AsDevice, ) -> Self

pub fn new_with_init( in_channels: usize, out_channels: usize, kernel_size: usize, config: impl IntoConv2dConfig, dtype: impl Type, device: impl AsDevice, weight_init: impl Init, bias_init: Option<impl Init>, ) -> Self

pub fn fwd(&self, x: &Tensor) -> Tensor

Trait Implementations

impl Clone for Conv2d

fn clone(&self) -> Conv2d

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Conv2d

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

Formats the value using the given formatter.

impl Module for Conv2d

type Input = Tensor

type Output = Tensor

fn forward(&self, input: &Self::Input) -> Self::Output

fn gather_params(&self, params: &mut HashMap<usize, Tensor>)

fn update_params(&self, params: &mut HashMap<usize, Tensor>)

fn gather_named_params( &self, prefix: &str, params: &mut HashMap<String, Tensor>, )

fn update_named_params( &self, prefix: &str, params: &mut HashMap<String, Tensor>, )

fn params(&self) -> HashMap<usize, Tensor>

fn named_params(&self, prefix: &str) -> HashMap<String, Tensor>

fn to_safetensors<P>(&self, filename: P)

where P: AsRef<Path>, Self: Sized,

fn update_by_safetensors<P>(&self, filenames: &[P], device: impl AsDevice)

where P: AsRef<Path>,

impl ValueSpec for Conv2d

type Kind = Module

type Tensors = HashMap<usize, Tensor>

type Gradient = HashMap<usize, Tensor>

impl TrainableModule for Conv2d