This dataset is generated as a part of a tutorial on fitting PyTorch models. The tutorial is at github.com/sdtemple/zootopia3.
You can generate your own data with answers/simulation-exercise.ipynb at the repo.
Treat this as held-out test data, with 20 examples for each shape (4) and color (8) combo.