cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_unflatten#

cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_unflatten(values, context)[source]#

After mapping the TensorSpecs to tensors/callables, return them directly as leaf nodes instead of reconstructing a dictionary when unflattening.

previous

cerebras.modelzoo.data.common.SyntheticDataProcessor.custom_dict_flatten

next

cerebras.modelzoo.data.common.SyntheticDataProcessor.SyntheticDataProcessor