使用PyTorch框架接口采集