TensorFlow:有没有办法将无类型的列表转换为Tensor？

Question

对于我的应用程序,我试图将列表转换[None, 1, 1, 64]为张量使用tf.convert_to_tensor([None, 1, 1, 64]),但这给了我错误:

TypeError: Failed to convert object of type <type 'list'> to Tensor. Contents: [None, 1, 1, 64]. Consider casting elements to a supported type.

理想情况下,我想None成为第一个维度,因为它代表了batch_size.目前,我唯一可以避免此错误的方法是明确地将batch_size提供给操作,但我希望有一种更简洁的方法将这样的列表转换为张量.

Answer 1

不,因为None并且64具有不同的类型,并且所有张量都是键入的:在一个张量中不能有不同类型的元素.

你最接近的事情是nan:

tf.convert_to_tensor([np.nan, 1, 1, 64])

虽然我无法想象为什么你会这么想.

但是你可以创建一个TensorShape:

tf.TensorShape([None, 1, 1, 64])