将 Flatten 层从 PyTorch 转换为 Tensorflow - 等效于 start_dim 和 end_dim

【问题标题】：Convert Flatten layer from PyTorch to Tensorflow - Equivalent for start_dim and end_dim将 Flatten 层从 PyTorch 转换为 Tensorflow - 等效于 start_dim 和 end_dim
【发布时间】：2021-10-11 23:03:24
【问题描述】：

Tensorflow 中 flatten PyTorch 层的选项 start_dim 和 end_dim 的等效项是什么？

对于 Tensorlfow，我们只有 data_format，而且它是不可定制的。

【问题讨论】：

在张量流的Flatten层中，总是start_dim=1和end_dim=-1。如果要展平其他维度，可以使用Reshape 层。红约重塑here.

标签： python tensorflow pytorch flatten

【解决方案1】：

我认为tf 中没有相同的实现。但是，您始终可以使用tf.reshape 并使用一个简单的函数自己添加形状，该函数将input、start_dim 和end_dim 作为参数，并输出torch.flatten 会给您的相应输出shape。

【讨论】：

非常感谢你们，我使用了重塑层！