For our tensor types we want to be able to (optionally) specify the shape of the tensor as part of the type: ```python Tensor[224, 224, 3] TorchTensor[128] ```
For our tensor types we want to be able to (optionally) specify the shape of the tensor as part of the type: