python 批量创建变量_python – TensorFlow：如何通过权重变量批处理批量张量？

我有以下批量形状：

[?,227,227]

以下权重变量：

weight_tensor = tf.truncated_normal([227,227],**{'stddev':0.1,'mean':0.0})

weight_var = tf.Variable(weight_tensor)

但是当我做tf.batch_matmul：

matrix = tf.batch_matmul(prev_net_2d,weight_var)

我失败了,出现以下错误：

ValueError: Shapes (?,) and () must have the same rank

所以我的问题变成：我该怎么做？

我如何在2D中获得一个乘以每个单独图片(227×227)的weight_variable,以便我有一个(227×227)输出？这个操作的扁平版本完全耗尽了资源…加上渐变不会在平面形状中正确改变重量……

或者：如何沿着批量维度(？,)分割传入的张量,以便我可以使用我的weight_variable在每个分割张量上运行tf.matmul函数？