archs/models.py [158:167]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    module_in_wt = module_in_wt.transpose(0, 1)
                    add_dims = prev_out[0].dim() + 1 - module_in_wt.dim()
                    module_in_wt = module_in_wt.view(*module_in_wt.shape,
                                                     *([1] * add_dims))
                    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 [179:188]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    module_in_wt = module_in_wt.transpose(0, 1)
                    add_dims = prev_out[0].dim() + 1 - module_in_wt.dim()
                    module_in_wt = module_in_wt.view(*module_in_wt.shape,
                                                     *([1] * add_dims))
                    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))
                    ])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -