Pytorch data inheriting namedtuples

data_namedtuple.py

from collections import namedtuple


class Data(namedtuple("Data", ("x", "y"))):
     def to(self, device, non_blocking=False):
         x = self.x.to(device, non_blocking=non_blocking)
         y = self.y.to(device, non_blocking=non_blocking)
         return Data(x, y)

     def contiguous(self):
         x = self.x.contiguous()
         y = self.y.contiguous()
         return Data(x, y)