archs/models.py [162:167]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    module_in_wt = module_in_wt.expand(
                        len(prev_out), *prev_out[0].shape)
                    module_in = sum([
                        module_in_wt[i] * prev_out[i]
                        for i in range(len(prev_out))
                    ])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



archs/models.py [202:207]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                        module_in_wt = module_in_wt.expand(
                            len(prev_out), *prev_out[0].shape)
                        module_in = sum([
                            module_in_wt[i] * prev_out[i]
                            for i in range(len(prev_out))
                        ])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -