import torchvision
from torchvision import datasets
+from _ext import flatland
+
######################################################################
def sequences_to_image(x):
######################################################################
-from _ext import flatland
-
-x = torch.ByteTensor()
-
-flatland.generate_sequence(10, x)
+x = flatland.generate_sequence(5, 3, 128, 96)
sequences_to_image(x).save('sequences.png')