Struct syntaxdot_transformers::layers::Conv1D

pub struct Conv1D {
    pub ws: Tensor,
    pub bs: Option<Tensor>,
    pub config: ConvConfig,
}

1-D convolution.

Fields

ws: Tensor

bs: Option<Tensor>

config: ConvConfig

Implementations

impl Conv1D

pub fn new<'a>(
    vs: impl Borrow<PathExt<'a>>,
    in_features: i64,
    out_features: i64,
    kernel_size: i64,
    groups: i64
) -> Result<Self, TransformerError>

Trait Implementations

impl Debug for Conv1D

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

Formats the value using the given formatter.

impl FallibleModule for Conv1D

type Error = TransformerError

The error type.

fn forward(&self, xs: &Tensor) -> Result<Tensor, Self::Error>

Apply the module.