def hybrid_forward()

in containers/Shoot/CNN/unet.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

    def hybrid_forward(self, F, x):
        pad_width=self.pad_width
        pad=(0,0,0,0,pad_width,pad_width,pad_width,pad_width)
        
        scaled_x=x
        padded=x.pad(mode="reflect",pad_width=pad)
        convs=self.down_branch(padded)
        up=self.up_branch(self.drop_out(convs[0]),convs[1:])
        
        up_cropped=up.slice_axis(
            axis=2,begin=pad_width,end=-pad_width
        ).slice_axis(
            axis=3,begin=pad_width,end=-pad_width
        )
        return self.activation(up_cropped)