block_diagonal#

torchdyno.models.initializers.block_diagonal(blocks)[source]#

Create a block diagonal matrix from a list of matrices.

Parameters:

  • blocks (torch.Tensor) – list of matrices.

  • couplings (Optional[List[Tuple[int, int, torch.Tensor]]], optional) – list of blocks coupling blocks in the diagonal. Defaults to None.

Returns:

block diagonal matrix.

Return type:

torch.Tensor