(self, x, skipx)
| 156 | self.ops = _make_nconv(spatial_dims, out_channels, nconvs, act) |
| 157 | |
| 158 | def forward(self, x, skipx): |
| 159 | if self.dropout is not None: |
| 160 | out = self.dropout(x) |
| 161 | else: |
| 162 | out = x |
| 163 | skipxdo = self.dropout2(skipx) |
| 164 | out = self.act_function1(self.bn1(self.up_conv(out))) |
| 165 | xcat = torch.cat((out, skipxdo), 1) |
| 166 | out = self.ops(xcat) |
| 167 | out = self.act_function2(torch.add(out, xcat)) |
| 168 | return out |
| 169 | |
| 170 | |
| 171 | class OutputTransition(nn.Module): |