For our tensor types we want to be able to (optionally) specify the shape of the tensor as part of the type:

Tensor[224, 224, 3]
TorchTensor[128]