Pytorch fold. functional as F z = F.