Python 在tensorflow中将张量调整为特定形状

我有一个形状的张量

（1,32,32,3）

。我想用shape

（16,8,8,3）

重塑它。有没有办法进行这种转换？

您可以使用：

---

只要你的张量中有相同数量的元素以整形形式存在，这个方法就有效。

如果你想做的是将张量从给定的

（1,32,32,3）

形状整形为

（16,8,8,3）

，可以使用

tf.reformate

。这是解释

>>> 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])