Caffeのレイヤー内のレイヤー

Question

私が書いたカスタム損失レイヤーがあります。このレイヤーは、下部[0]ブロブの一部にsoftmaxとsigmoidのアクティブ化を適用します。

Ex: `bottom[0]` is of shape (say): `[20, 7, 7, 50]` (`NHWC` format) 
I would like to apply `softmax` to `[20, 7, 7, 25]` (first 25 channels) and 
`sigmoid` to `[20, 7, 7, 1]` (just one channel) and the remaining 24 channels are taken in as it is. 

にはどうすれば効果的にこれらの二つのsoftmaxとsigmoid層の入力塊にメモリを割り当て、このメモリを解放するのですか？

Answer 1

代わりの内部データを割り当て、あなたは単に外部"Slice"層を使用し、カフェ「既製」レイヤーを使用して入力ブロブをスライスすることができます

layer { 
    name: "slice" 
    type: "Slice" 
    bottom: "input_to_loss" 
    top: "to_softmax" 
    top: "to_sigmoid" 
    top: "leftovers" 
    slice_param { 
    axis: -1 # slice the last axis 
    slice_point: 25 
    slice_point: 26 
    } 
} 

Answer 2

ネットワークの入出力ブロブと一緒に、すべての中間アクティベーションはネットクラスによって管理され、Net<Dtype>::Initファンクション（src/caffe/net.cpp）で設定されます。

レイヤー内から上部と下部のブロブメモリを割り当てる/割り当てを解除する必要はありません。