| 108 | |
| 109 | |
| 110 | class ScipyConv2d(Module): |
| 111 | def __init__(self, filter_width, filter_height): |
| 112 | super(ScipyConv2d, self).__init__() |
| 113 | self.filter = Parameter(torch.randn(filter_width, filter_height)) |
| 114 | self.bias = Parameter(torch.randn(1, 1)) |
| 115 | |
| 116 | def forward(self, input): |
| 117 | return ScipyConv2dFunction.apply(input, self.filter, self.bias) |
| 118 | |
| 119 | |
| 120 | ############################################################### |
no outgoing calls
no test coverage detected