Function get_binary_kernel

monai/networks/layers/simplelayers.py:430–438

Create a binary kernel to extract the patches. The window size HxWxD will create a (H*W*D)xHxWxD kernel.

(window_size: Sequence[int], dtype=torch.float, device=None)

def get_binary_kernel(window_size: Sequence[int], dtype=torch.float, device=None) -> torch.Tensor:
    """
    Create a binary kernel to extract the patches.
    The window size HxWxD will create a (H*W*D)xHxWxD kernel.
    """
    win_size = convert_to_tensor(window_size, int, wrap_sequence=True)
    prod = torch.prod(win_size)
    s = [prod, 1, *win_size]
    return torch.diag(torch.ones(prod, dtype=dtype, device=device)).view(s)  # type: ignore


def median_filter(