python - Tensorflow 中是否有任何方法可以从 4D 向量制作 2D 向量？

Question

我有一个维度为 [n, 56202] 的输入向量。

由于我的神经网络的第一层，我想将我的向量变成 [n, 238, 238]。

是否有任何方法（ex）tf.matmul）可以在 Tensorflow 中将 [n,56202] 变为 [n, 238, 238] ？

我尝试如下，但没有奏效。

placeholder_X = tf.compat.v1.placeholder(tf.float32, shape=[None, 56202])
W0 = tf.Variable(tf.random.normal([56202, 238, 238, 1], mean=0, stddev=0.01), name='W0')
Z0 = tf.matmul(placeholder_X, W0)

标签： pythontensorflowmatrixmatrix-multiplicationvector-multiplication