我有以下批量形状:
[?,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函数?