サイズの異なる行列の集合をパディングしてサイズを揃えたい
以下のように、サイズの異なる行列の集合Aが与えられた時、
A =
[[[1, 2, 3],
[1, 2, 3],
[1, 2, 3]],
[[1, 2],
[1, 2],
[1, 2],
[1, 2]],
[[1, 2, 3, 4],
[1, 2, 3, 4]]]
以下のように、0でパディングして全体を連結させて、1つの行列にしたいです。
B =
[[1, 2, 3, 0],
[1, 2, 3, 0],
[1, 2, 3, 0],
[1, 2, 0, 0],
[1, 2, 0, 0],
[1, 2, 0, 0],
[1, 2, 0, 0],
[1, 2, 3, 4],
[1, 2, 3, 4]]
Chainerで上のような入力と出力を実現するためには、
どのようなコードを書けば良いでしょうか?
AとBはnumpy array型です。