def METHOD_NAME(self, Wh):
    N = Wh.size()[0]
    Wh_repeated_in_chunks = Wh.repeat_interleave(N, dim=0)
    Wh_repeated_alternating = Wh.repeat(N, 1)
    all_combinations_matrix = torch.cat(
        [Wh_repeated_in_chunks, Wh_repeated_alternating], dim=1
    )
    return all_combinations_matrix.view(N, N, 2 * self.out_features)