假设我在Tensorflow中有一个形状[3,4,4]
的张量,该张量是未知的。如何将其形状更改为[3,4,4,1]
?
我尝试使用重塑函数,但由于我不明确知道张量形状,因此无法直接使用它。
答案 0 :(得分:1)
鉴于形状为[3,4,4]
的t,可以将其重塑为[3,4,4, 1]
tf.reshape(t, [3,4,4, 1]) # by explicitly specifying the shape
tf.expandims(t, 3) # implicitly we expand the dimension 3 with size 1