def METHOD_NAME(self):
    all_scales = []
    for dense_scale, qkv_scale, m4hh_scale, mh4h_scale in \
        zip(self.dense_scales, self.qkv_scales, self.mlp4hh_scales, self.mlph4h_scales):
        all_scales.append(self.merge_layer_scales([qkv_scale, dense_scale, mh4h_scale, m4hh_scale]))
    return torch.cat(all_scales)