result comes from here
y = self.atom_pool(yi, atom_mask)
result = [y]
props = ['y']

at_func = namedtuple('atomwise', props)
return at_func(*result)


as for the batch is just a tensor of a given batch_size created from the Dataloader.
I feel, anyway, that the error might be related to how result is created.