cuvis_ai.utils.torch.get_output_shape

cuvis_ai.utils.torch.get_output_shape(input_shape, model)[source]