block_diagonal_coupling#

torchdyno.models.initializers.block_diagonal_coupling(block_sizes, couplings)[source]#

Create the coupling matrix for a given block diagonal matrix.

Parameters:

  • block_sizes (List[int]) – list of block sizes.

  • couplings (List[Tuple[int, int, torch.Tensor]]) – list of coupling blocks.

Returns:

coupling matrix

Return type:

torch.Tensor