Introduction
Get Started
Prepare your data
Cerebras PyTorch API
cerebras.pytorch
cerebras.pytorch.amp
cerebras.pytorch.optim
cerebras.pytorch.sparse
cerebras.pytorch.metrics
Cerebras Model Zoo
Cerebras Guides
Fundamentals
Support
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