That is very weird. Maybe you should post a snippet of the code (if possible) so that we can figure out where the problem is.  I honestly don’t think that the torch.zeros function was changed so drastically as to cause that worsening in performance with the update. I could be wrong tho.