vaibkumr/nTensor6.py

Created October 11, 2019 17:37

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/vaibkumr/c43b8d169ac5c85be4783380bed34ad1.js"></script>
Save vaibkumr/c43b8d169ac5c85be4783380bed34ad1 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

nTensor6.py

	import torch

	# Unary operator on named tensors
	t = torch.randn(4, 2, names=('N', 'C'))
	t = t.abs()
	t.names #output: ('N', 'C')

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