Constructs TensorSpec instances to contain the leaf nodes of the tree structure before flattening.
previous
cerebras.modelzoo.data.common.SyntheticDataProcessor
next
cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_unflatten