pub struct Conv2d {
pub weight: Tensor,
pub stride: Vec<i64>,
pub padding: Padding,
pub dilation: Vec<i64>,
pub groups: i64,
pub bias: Tensor,
pub input_shape: Option<Vec<usize>>,
pub metadata: MetadataMap,
}Expand description
2-D convolution.
Fields§
§weight: TensorKernel weights, typically (C_out, C_in, Kh, Kw).
stride: Vec<i64>Stride per spatial axis (or single value expanded later).
padding: PaddingPadding specification.
dilation: Vec<i64>Dilation per spatial axis.
groups: i64Grouped convolution groups.
bias: TensorBias of shape (C_out,) (required on the NIR wire).
input_shape: Option<Vec<usize>>Optional spatial input (N_x, N_y).
metadata: MetadataMapFree-form node metadata (Python metadata dict).
Trait Implementations§
impl StructuralPartialEq for Conv2d
Auto Trait Implementations§
impl Freeze for Conv2d
impl RefUnwindSafe for Conv2d
impl Send for Conv2d
impl Sync for Conv2d
impl Unpin for Conv2d
impl UnsafeUnpin for Conv2d
impl UnwindSafe for Conv2d
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more