pytorch做clip