TensorFlow:有没有办法将 None 类型的列表转换为 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])