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