- 博客(1)
- 收藏
- 关注
原创 关于tensorflow和pytorch代码转换的大坑,记录一下
tf和torch都有.spilt函数tf.spilt中第二个参数 是你想要把这一维分成几份torch中第二个参数 是该维切分之后每份有多少个因为代码前半部分一直是100要分10份,所以一直没报错,我以为这俩能互替了。结果后边不是100了,怎么分都不对,还没发现是这问题,折磨太久了
2022-09-02 07:11:07
458
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人