switch pytorch code

切换到pytorch代码块，而不是默认的块，仅在第一次进入界面时生效

这些是代码更新过的版本。显示所有版本