gpt4 book ai didi

python - Keras 的功能类似于 Torch 中的 View

转载 作者:太空宇宙 更新时间:2023-11-03 14:56:38 26 4
gpt4 key购买 nike

我正在努力将 Torch 和 lua 中的一些代码转换为 Keras 和 python。 Keras 中是否有一个功能类似于 View在 torch ? 编辑:我想使用该函数将卷积层的输出转换为正确的形状,作为全连接层的输入。

最佳答案

我确实得到了答案。我可以使用Reshape 。但是,对于我的情况,我使用 Flatten在传递到完全连接层之前“展平”形状。

关于python - Keras 的功能类似于 Torch 中的 View,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45513000/

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