_sesn_valid_mask_on_residual_stream

mosfit.modules._sesn_valid_mask_on_residual_stream(kwargs, residual_n)[source]

valid_mask is dense like observed; diagonal uses mask[observed].