pytorch 与 tensorflow 2.0 的视图等价是什么？

Question

l_conv7 = self.loc_conv7(conv7_feats)  # (N, 24, 19, 19)
l_conv7 = l_conv7.permute(0, 2, 3, 1).contiguous()  # (N, 19, 19, 24)
l_conv7 = l_conv7.view(batch_size, -1, 4)  # (N, 2166, 4), there are a total 2116 boxes on this feature map

在 TensorFlow 中与火炬视图的等价性是什么？如何更改 TensorFlow 2.0 中的 l_conv7.view？

Answer 1

用

l_conv7.reshape(batch_size, -1, 4)