cuvis_ai.utils.torch.guess_input_dimensionalty

cuvis_ai.utils.torch.guess_input_dimensionalty(model: Module) → InputDimension[source]

Inspects the first layer of a PyTorch model to guess if the model expects 1D (flattened), 2D (spatial), or 3D input data.