create_strategy

deepinv.distributed.strategies.create_strategy(img_size, tiling_dims=None, **kwargs)

Create a distributed signal strategy.

Parameters:

• img_size (Sequence[int]) – full shape of the signal tensor, including batch and channel dimensions (e.g., (B, C, H, W)).

• tiling_dims (tuple[int, ...] | None) – dimensions to tile. If None, defaults to last N dimensions.

Returns:

the created strategy instance.

Return type:

DistributedSignalStrategy