Are there any PyTorch dataset attributes which I could use for shape detection?

I want to make a shape detection AI so I want to know if there are any PyTorch dataset attributes like MNIST which I could use for making it. Therefore, the dataset attribute must
have shapes for my AI.

I do not need an answer anymore so please do not reply to this question.