Skip to content

Instantly share code, notes, and snippets.

@vaibkumr

vaibkumr/nTensor2.py

Created October 11, 2019 16:32
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save vaibkumr/135c720a2133b334e5309fcaa8eea177 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save vaibkumr/135c720a2133b334e5309fcaa8eea177 to your computer and use it in GitHub Desktop.
Download ZIP
none
Raw
nTensor2.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