mhh strange I tested my input and at various stage of the forward and every time I have something different 
EDIT: the only things which could lead to a problem on my input are :
np.expand_dims
torch.from_numpy
j.permute
otherwise I don’t see what could insert randomness in this model