哈工大版本,for pytorch