Module fl_layers.utils

Expand source code
#!/usr/bin/env python3

# Import Tensor node in the computation graph
from torch import Tensor


def input_to_block_diag(x_array: Tensor) -> Tensor:
    pass

Functions

def input_to_block_diag(x_array: torch.Tensor) ‑> torch.Tensor
Expand source code
def input_to_block_diag(x_array: Tensor) -> Tensor:
    pass