Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/dimensions.py

Created August 24, 2019 17:48
Show Gist options
  • Save omarsar/d9c3ecec714b1ba315177bb6c86908b4 to your computer and use it in GitHub Desktop.
Save omarsar/d9c3ecec714b1ba315177bb6c86908b4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dimensions.py
for images, labels in trainloader:
print("Image batch dimensions:", images.shape)
print("Image label dimensions:", labels.shape)
break
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment