torch >= 1.3