Attend the Women in ML Symposium on December 7 Register now


Stay organized with collections Save and categorize content based on your preferences.

Adds an outer dimension to the shape of input specs.

Used in the notebooks

Used in the tutorials

specs Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
dim Int, representing the outer dimension to be added to the TensorSpecs in specs.

Nested TensorSpecs with outer dimensions added to the shape of input specs.