pytorch 的视图与 tensorflow 2.0 的等效性是什么?
what is the pytorch's view equivalence with tensorflow 2.0?
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 中的 torch's view 有什么等价性?
如何更改 TensorFlow 2.0 中的 l_conv7.view?
使用
l_conv7.reshape(batch_size, -1, 4)
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 中的 torch's view 有什么等价性? 如何更改 TensorFlow 2.0 中的 l_conv7.view?
使用
l_conv7.reshape(batch_size, -1, 4)