pytorch部署C++时出现如下问题该如何解决

你看自己下面注释掉的那行？【1，3，256，256】，你写成【1，256，256，3】了