Skip to content

Instantly share code, notes, and snippets.

@vaibkumr

vaibkumr/nTensor2.1.py

Created October 11, 2019 16:41
Show Gist options
  • Save vaibkumr/a109f006f9422b92aa45c48d0a403220 to your computer and use it in GitHub Desktop.
Save vaibkumr/a109f006f9422b92aa45c48d0a403220 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nTensor2.1.py
import torch
batch = torch.zeros(64, 3, 100, 100, names=('N', 'C', 'H', 'W'))
print(batch.shape) #torch.Size([64, 3, 100, 100])
batch = batch.align_to('N', 'H', 'W', 'C')
print(batch.shape) #torch.Size([64, 100, 100, 3])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment