使用libtorch完成语义分割