Method random_idx

monai/utils/ordering.py:193–207

(rows: int, cols: int, depths: int | None = None)

    @staticmethod
    def random_idx(rows: int, cols: int, depths: int | None = None) -> np.ndarray:
        idx: list[tuple] = []

        for r in range(rows):
            for c in range(cols):
                if depths:
                    for d in range(depths):
                        idx.append((r, c, d))
                else:
                    idx.append((r, c))

        idx_np = np.array(idx)
        np.random.shuffle(idx_np)

        return idx_np