Variable変数の配列の操作について

Question

Variable変数の配列の操作についての質問です。

予測値を計算した際に（例えば Model.prediction）、以下のような
Variable変数の配列が出力されると思います。
z = variable([[a1]
[a2]
[b1]
[b2]
[c1]
[c2]])
これを以下の様に操作して得られる３つの要素が参照値と合うように
更新されるタイプのコードを作成しています。
y = F.sum( z.reshape(3,2), axis = 1 ).reshape(3,1)
= variable([[a1+a2]
[b1+b2]
[c1+c2]])

上の例の様にreshapeで上手く取り扱える場合はいいのですが、下の
ように数が合わない場合の取り扱いに苦慮しています。
z = variable([[a1]
[a2]
[b1]
[b2]
[c1]])

思った形になるように無理やり（Variable変数をスカラー変数にし、
再度戻したりして）計算してみましたが、ネットワークは更新され
ませんでした。

おそらくVariable変数がネットワークの構造を保持しており、それ
を切るような無理やりな計算をしたためだと推測しています。上に
書いたようなreshpaeとF.sumであれば、その情報が途切れないのか
なと考えています。

アイデアございましたらご教示いただけると幸いです。