TensorFlow: Is there a way to convert a list with None type to a Tensor?

Viewed 19063

For my application, I am trying to convert a list with [None, 1, 1, 64] to a tensor using tf.convert_to_tensor([None, 1, 1, 64]), but this gives me the error:

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

Ideally, I want None to be the first dimension because it represents the batch_size. Currently, the only way I could avoid this error is to explicitly give the batch_size to the operation, but I am hoping there is a cleaner way to convert such a list to a tensor.

2 Answers

Use tf.convert_to_tensor([-1, 1, 1, 64]) instead of None, since you are already specifying 3 out of 4 dimensions, this should be fine.

Related