numpyを使って要素数の違う多次元配列の初期化の仕方

Question

前提・実現したいこと

ニューラルネットワークを作成している中で、重みの配列を作成しようとしたのですが、
要素数の違う多次元配列の初期化の仕方がわからなく、質問いたしました。

ex) w = [ { (0 0 0)(0 0 0) }{ (0 0)(0 0)(0 0) } ]

このように3x2の配列と2*3の配列を一つの配列で初期化したいです。
見やすくするためにあえて括弧の形を変えています。

試したこと

コード

import numpy as np

w = np.random.rand(2,3,2)

print(w)

結果

[[[0.76085332 0.94623838]
  [0.31026028 0.91528297]
  [0.33379404 0.21865818]]

 [[0.04586849 0.06809018]
  [0.08647367 0.94940481]
  [0.63085373 0.62089273]]]

どちらも3x2の配列ですが、それを任意の2次配列にしたいです。

追記

配列の要素数の指定は変数で行います。
一度に初期化できる方法を探しているので、二つの2次配列をそれぞれ初期化して組み合わせるなどはNGです。

Answer 1

やりたいことはこういうことでしょうか？

import numpy as np
w = np.random.rand(2,3,2)

w1 = w[0].reshape(3,2) # 3x2
print(w1)
"""
[[0.26954749 0.01297769]
 [0.10067051 0.3162785 ]
 [0.19149429 0.28691526]]
"""
w2 = w[1].reshape(2,3) # 2x3
print(w2)
"""
[[0.23993565 0.51013366 0.54704672]
 [0.81432386 0.69813558 0.6853153 ]]
"""

# 変数でも可
M,N = (2,3)
w3 = w[0].reshape(M,N) # MxN