(self, x)
| 781 | self.bias = bias |
| 782 | |
| 783 | def initialize(self, x): |
| 784 | self.in_channels = x.shape[1] |
| 785 | self.depthwise_conv = Conv2d( |
| 786 | self.in_channels, |
| 787 | self.kernel_size, |
| 788 | stride=self.stride, |
| 789 | padding=self.padding, |
| 790 | group=self.in_channels, |
| 791 | bias=self.bias, |
| 792 | ) |
| 793 | |
| 794 | self.point_conv = Conv2d(self.nb_kernels, 1, bias=self.bias) |
| 795 | |
| 796 | def forward(self, x): |
| 797 | y = self.depthwise_conv(x) |