| 3 | from .FlowProjectionLayer import FlowProjectionLayer #, FlowFillholeLayer |
| 4 | |
| 5 | class FlowProjectionModule(Module): |
| 6 | def __init__(self, requires_grad = True): |
| 7 | super(FlowProjectionModule, self).__init__() |
| 8 | |
| 9 | self.f = FlowProjectionLayer(requires_grad) |
| 10 | |
| 11 | def forward(self, input1): |
| 12 | return self.f(input1) |
| 13 | |
| 14 | # class FlowFillholeModule(Module): |
| 15 | # def __init__(self,hole_value = -10000.0): |
no outgoing calls