如何重塑维度未知的张量？答案

【问题标题】：How to reshape a tensor with unknown dimensions?如何重塑维度未知的张量？
【发布时间】：2023-01-19 19:40:21
【问题描述】：

例如，我有一个形状为 (None, None, 3) 的张量，其中 1-st None 代表批量大小，而 2-nd None 只是一个未知维度。让我们假设二维实际上是一些数字x。所以张量的形状是(batch_size, x, 3)，其中batch_size和x是未知的。如何在没有给定 x 的情况下将该张量重塑为 (batch_size, x * 3) 的形状？

请注意，这个问题不同于this one。在我的例子中，它不仅仅是一个具有 None 维度的张量，在我的例子中，None 维度本身应该在重塑后减少。

【问题讨论】：

标签： tensorflow keras reshape tensor tf.keras

【解决方案1】：

在 tensorflow reshape 文档中，您可以阅读：

如果形状的一个分量是特殊值 -1，则该分量的大小计算维度以使总大小保持不变。

所以如果你知道 batch_size 你可以重塑为 (batch_size,-1)

【讨论】：