def run_variable_channel_forward()

in models/swin_transformer_3d.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

    def run_variable_channel_forward(self, x):
        sidx = 0
        out = None
        for idx in range(len(self.additional_variable_channels)):
            eidx = sidx + self.additional_variable_channels[idx]
            c_out = self.var_proj[idx](x[:, sidx:eidx, ...])
            if idx == 0:
                out = c_out
            else:
                out += c_out
            sidx = eidx
        return out