shape [64, 256，256] is invalid for input of size。。。

shape [64, 256，256]] is invalid for input of size错误我的模型在输入张量的时候，出现了shape [64, 256，256]] is invalid for input of size错误，这种错误，往往是跑pycharm在核实张量元素的时候，发现我们设置的张量维度和实际的数据不符合出现的，只需要在高维度处加-1即可。-1表示此处的维度随程序自动变化

☆ﾐ(o*･ω･)ﾉ

42090人浏览 · 2021-02-22 17:33:47

☆ﾐ(o*･ω･)ﾉ · 2021-02-22 17:33:47 发布

shape [64, 256，256]] is invalid for input of size错误

我的模型在输入张量的时候，出现了shape [64, 256，256]] is invalid for input of size错误，这种错误，往往是跑pycharm在核实张量元素的时候，发现我们设置的张量维度和实际的数据不符合出现的，只需要在高维度处加-1即可。-1表示此处的维度随程序自动变化。可以很好的解决问题
在高维出加-1即可