Method
depthwise_conv1d_bias_ctx
(
&self,
ctx: &Tensor,
input: &Tensor,
weight: &Tensor,
bias: &Tensor,
kernel_size: usize,
channels: usize,
)
Source from the content-addressed store, hash-verified
| 236 | } |
| 237 | |
| 238 | fn depthwise_conv1d_bias_ctx( |
| 239 | &self, |
| 240 | ctx: &Tensor, |
| 241 | input: &Tensor, |
| 242 | weight: &Tensor, |
| 243 | bias: &Tensor, |
| 244 | kernel_size: usize, |
| 245 | channels: usize, |
| 246 | ) -> Result<Tensor> { |
| 247 | // Concatenate [ctx, input] along time dimension, then run conv |
| 248 | let combined = Tensor::cat(&[ctx, input], 2)?; |
| 249 | self.depthwise_conv1d_bias(&combined, weight, bias, kernel_size, channels) |
| 250 | } |
| 251 | |
| 252 | fn add3(&self, a: &Tensor, b: &Tensor, c: &Tensor) -> Result<Tensor> { |
| 253 | (a + b)? + c |