以下代码源自dive into DL T2.0, 运行时间较长，建议在colab上运行。 ...

...