operator-keyword - Caffe2でnumpyのような「収集」操作を適用するには?

Question

私は Caffe2 が初めてで、次のような操作を作成したいと考えています。

• ナンピーウェイ

サンプルコード

• ピトックウェイ

サンプルコード

私の質問は、上記のような同じ演算子を作成するために Caffe2 演算子を構成する方法ですか? いくつかのコンポジションを試しましたが、それでも正しいコンポジションを見つけることができませんでした。組成を知っている人がいたら、助けてください。本当に感謝します。