nn-varfloat-pytorch