tf和torch都有.spilt函数
tf.spilt中第二个参数 是你想要把这一维分成几份
torch中第二个参数 是该维切分之后每份有多少个
因为代码前半部分一直是100要分10份,所以一直没报错,我以为这俩能互替了。
结果后边不是100了,怎么分都不对,还没发现是这问题,折磨太久了
关于tensorflow和pytorch代码转换的大坑,记录一下
最新推荐文章于 2023-07-11 16:53:34 发布
![](https://img-home.csdnimg.cn/images/20240711042549.png)
tf和torch都有.spilt函数
tf.spilt中第二个参数 是你想要把这一维分成几份
torch中第二个参数 是该维切分之后每份有多少个
因为代码前半部分一直是100要分10份,所以一直没报错,我以为这俩能互替了。
结果后边不是100了,怎么分都不对,还没发现是这问题,折磨太久了