gpt4 book ai didi

tensorflow - pytorch 的 View 与 tensorflow 2.0 的等效性是什么?

转载 作者:行者123 更新时间:2023-12-01 21:34:01 25 4
gpt4 key购买 nike

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 View 有何等价关系?如何更改 TensorFlow 2.0 中的 l_conv7.view?

最佳答案

使用

l_conv7.reshape(batch_size, -1, 4)

关于tensorflow - pytorch 的 View 与 tensorflow 2.0 的等效性是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62185282/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com