gpt4 book ai didi

machine-learning - caffe LayerSetUp 和 Reshape?

转载 作者:行者123 更新时间:2023-11-30 08:51:14 25 4
gpt4 key购买 nike

我现在正在阅读caffe源代码。
我对 LayerSetUpReshape 方法感到困惑。
有些层同时具有这两种方法,其他层只有一种或没有......为什么?谁能给我解释一下吗?

最佳答案

LayerSetUp 在加载网络时调用一次。其目标是
(a) 验证该层具有正确数量的输入/输出 blob
(b) 从prototxt中读取该层的参数
(c) 初始化内部参数

另一方面,Reshape 用于为参数和输出 blob 分配内存,甚至可以在网络设置后调用。例如,检测网络通常会更改输入形状,从而 reshape 所有后续的 blob。

关于machine-learning - caffe LayerSetUp 和 Reshape?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35100458/

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