test.py

   1 #!/usr/bin/env python-for-pytorch
   2
   3 import torch
   4 import torchvision
   5 from torchvision import datasets
   6
   7 from _ext import mylib
   8
   9 x = torch.ByteTensor(4, 5).fill_(0)
  10
  11 print(x.size())
  12
  13 mylib.generate_sequence(8, x)
  14
  15 print(x.size())
  16
  17 x = x.float().sub_(128).div_(128)
  18
  19 for s in range(0, x.size(0)):
  20     torchvision.utils.save_image(x[s], 'example_' + str(s) + '.png')