| 3 | from .InterpolationChLayer import InterpolationChLayer |
| 4 | |
| 5 | class InterpolationChModule(Module): |
| 6 | def __init__(self,ch): |
| 7 | super(InterpolationChModule, self).__init__() |
| 8 | self.ch = ch |
| 9 | # self.f = InterpolationChLayer(ch) |
| 10 | |
| 11 | def forward(self, input1, input2): |
| 12 | return InterpolationChLayer.apply(input1, input2) |
| 13 | |
| 14 | #we actually dont need to write the backward code for a module, since we have |
| 15 |
no outgoing calls