如何在python中更改张量形状?

时间:2019-02-26 05:46:02

标签: python tensorflow

假设我在Tensorflow中有一个形状[3,4,4]的张量,该张量是未知的。如何将其形状更改为[3,4,4,1]
我尝试使用重塑函数,但由于我不明确知道张量形状,因此无法直接使用它。

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