附录B PyTorch示例代码