tensorFlowtf.reshape()报错信息 - TypeError: Expected binary or unicode string

Posted 2020-12-04 japyc180717

今天在使用tensoflow跑cifar10的CNN分类时候，download一个源码，但是报错

TypeError: Failed to convert object of type <class ‘list‘> to Tensor. Contents: [-1, Dimension(4608)]. Consider casting elements to a supported type.

跟踪发现是tf.reshape()时候报错！

1 flatten_shape = input.get_shape()[1] * input.get_shape()[2] * input.get_shape()[3]
2 return tf.reshape(input, [-1, flatten_shape], name="flatten")

这里需要改成

flatten_shape = input.get_shape().as_list()[1] * input.get_shape().as_list()[2] * input.get_shape().as_list()[3]
return tf.reshape(input, [-1, flatten_shape], name="flatten")

需要使用.as_list()将获取到的shape转换成list才行。