Method __init__

doctr/transforms/modules/pytorch.py:192–193

(self)

    """Randomly shuffle channel order of a given image"""

    def __init__(self):
        super().__init__()

    def forward(self, img: torch.Tensor) -> torch.Tensor:
        # Get a random order