MCPcopy Create free account
hub / github.com/evilsocket/cake / conv1d

Method conv1d

cake-core/src/backends/mod.rs:610–628  ·  view source on GitHub ↗
(
        &self,
        x: &Tensor,
        weight: &Tensor,
        bias: Option<&Tensor>,
        padding: usize,
        stride: usize,
        dilation: usize,
        groups: usize,
    )

Source from the content-addressed store, hash-verified

608 /// Input: `(batch, in_channels, length)`, weight: `(out_channels, in_channels/groups, kernel_size)`.
609 #[allow(clippy::too_many_arguments)]
610 fn conv1d(
611 &self,
612 x: &Tensor,
613 weight: &Tensor,
614 bias: Option<&Tensor>,
615 padding: usize,
616 stride: usize,
617 dilation: usize,
618 groups: usize,
619 ) -> Result<Tensor> {
620 let out = x.conv1d(weight, padding, stride, dilation, groups)?;
621 match bias {
622 Some(b) => {
623 let b = b.reshape((1, b.dim(0)?, 1))?;
624 out.broadcast_add(&b)
625 }
626 None => Ok(out),
627 }
628 }
629
630 /// Transposed 1D convolution: `conv_transpose1d(x, weight) + bias`.
631 /// Matches candle_nn::ConvTranspose1d::forward() semantics.

Callers 10

conv1dFunction · 0.80
encodeMethod · 0.80
encode_streamingMethod · 0.80
forwardMethod · 0.80
decodeMethod · 0.80
decode_streamingMethod · 0.80

Implementers 5

mod.rscake-core/src/backends/rocm/mod.rs
mod.rscake-core/src/backends/cpu/mod.rs
mod.rscake-core/src/backends/metal/mod.rs
mod.rscake-core/src/backends/vulkan/mod.rs
mod.rscake-core/src/backends/cuda/mod.rs

Calls

no outgoing calls