Struct Conv2d

pub struct Conv2d { /* private fields */ }

Implementations

impl Conv2d

pub fn new(weight: Tensor, bias: Option<Tensor>, config: Conv2dConfig) -> Self

pub fn config(&self) -> &Conv2dConfig

pub fn weight(&self) -> &Tensor

pub fn bias(&self) -> Option<&Tensor>

pub fn absorb_bn(&self, bn: &BatchNorm) -> Result<Self>

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

fn forward(&self, x: &Tensor) -> Result<Tensor>