在 tensorflow 中將張量調整為特定形狀

Question

我有一個形狀為(1,32,32,3)的張量。 我想用形狀(16,8,8,3)重塑它。 有沒有辦法進行這種轉換？

Answer 1

您可以使用tf.reshape ：

>>> a = tf.random.uniform((1,32,32,3))
>>> a.shape
TensorShape([1, 32, 32, 3])
>>> b = tf.reshape(a,(16,8,8,3))
>>> b.shape
TensorShape([16, 8, 8, 3])

只要您的張量中以重新整形的形式存在相同數量的元素，這就會起作用。

Answer 2

如果您想要做的是將張量從(1,32,32,3)的給定形狀重塑為(16,8,8,3) ，則可以使用tf.reshape來完成。 它在這里解釋。