neuralop.data.datasets.load_darcy_flow_small

neuralop.data.datasets.load_darcy_flow_small(n_train, n_tests, batch_size, test_batch_sizes, data_root=PosixPath('/home/runner/work/neuraloperator/neuraloperator/neuralop/data/datasets/data'), test_resolutions=[16, 32], encode_input=False, encode_output=True, encoding='channel-wise', channel_dim=1)[source]