cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_flatten#

cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_flatten(d)[source]#

Constructs TensorSpec instances to contain the leaf nodes of the tree structure before flattening.