tensorplay._shape_funcs.block_diag

tensorplay._shape_funcs.block_diag(*tensors)[source]

Builds a block diagonal matrix from the given blocks.

0-D blocks become 1x1 matrices and 1-D blocks become single rows. The blocks are promoted to one type and must share a device; the zeros around them are made on that device by the operator itself.