monk1337/check_tensor_size.py

Created June 7, 2018 15:35

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/monk1337/77f2dba0e0ede8eba077379e6dc74652.js"></script>
Save monk1337/77f2dba0e0ede8eba077379e6dc74652 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

check_tensor_size.py

	print(tensor.size())

	# output:
	# torch.Size([4, 3])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment