tf.nn.conv2d(input,filter,[1, 1, 1, 1])要求input是一个四维张量(即 batch,heigh,width,chanel),但由于数据是由如下代码得到的:
training_dataset = dataset.batch(n_batch).prefetch(64) # n_batch=64
training_iterator = training_dataset.make_one_shot_iterator()
training_batch = training_iterator.get_next() # (?,64,64,3)dtype=tf.float32
此时的张量training_batch的shape为(?,64,64,3),接下来我想对training_batch使用卷积操作,但他的batch维是“?”无法使用tf.nn.conv2d()。
请问这个问题应该如何解决?