Skip to content

Instantly share code, notes, and snippets.

@sachadee

sachadee/tensorinfo.py

Last active May 7, 2022 06:55
Show Gist options
  • Save sachadee/ab04f05720caec73548e7033624c9013 to your computer and use it in GitHub Desktop.
Save sachadee/ab04f05720caec73548e7033624c9013 to your computer and use it in GitHub Desktop.
Download ZIP
tensor info
Raw
tensorinfo.py
#Base Image
print("Image Dtype:", np.array(image).dtype)
print("Image shape:", np.array(image).shape)
print('***********')
#Tensor image
print("Tensor Dtype:", tensor.dtype)
print("Tensor shape:", tensor.shape)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment