(self, inp, weight, bias)
| 564 | return (1, self.out_channels, 1, 1, 1) |
| 565 | |
| 566 | def calc_conv(self, inp, weight, bias): |
| 567 | return conv3d( |
| 568 | inp, |
| 569 | weight, |
| 570 | bias, |
| 571 | self.stride, |
| 572 | self.padding, |
| 573 | self.dilation, |
| 574 | self.groups, |
| 575 | self.conv_mode, |
| 576 | ) |
| 577 | |
| 578 | def forward(self, inp): |
| 579 | return self.calc_conv(inp, self.weight, self.bias) |