grain.experimental.get_element_spec

grain.experimental.get_element_spec(ds)

Returns specification of the elements produced by this dataset.

Does not instantiate iterator, perform any data reads or transformations.

Parameters:

ds (MapDataset | IterDataset) – MapDataset or IterDataset to get the element spec from.

Return type:

Any