cerebras.modelzoo.data.common.tensor_spec.TensorSpec#

class cerebras.modelzoo.data.common.tensor_spec.TensorSpec(**kwargs)[source]#

Bases: object

Wrapper class used to wrap the leaf nodes in SyntheticDataProcessor’s input.

TensorSpecs hold a dictionary of arguments used to specify a tensor. An instance of this class is constructed to wrap a dictionary if the dictionary in the input contains at least one of ‘shape’, ‘dtype’, or ‘tensor_factory’ keys.

Example list element format in yaml file:

shape: … dtype: …

This class merely holds the provided dictionary of kwargs. See models/common/pytorch/input/SyntheticDataProcessor.py for more docs and use cases.

Parameters
  • kwargs – Any variable number of keyword arguments written as a dictionary

  • above. (under the tag in the .yaml file as seen in the example) –

Methods