1

カスタムsoftmax関数を使用しています。xゼロの新しいテンソルの形状の要素としてテンソルの形状を使用しようとしています。intではないのでできません。

def custom_softmax(x):
    sh = K.shape(x)
    ...      
    xc = K.zeros((sh[0] * 16 * 16, 1))
    ...

私が試した次のオプションは、機能するはずのテンソルの評価です。

def custom_softmax(x):
    sh = K.shape(x)
    sess = K.get_session()
    ...      
    xc = K.zeros((sh[0].eval(session=sess) * 16 * 16, 1))
    ...

それは私にエラーを与える

tensorflow.python.framework.errors_impl.InvalidArgumentError: You must feed a value for placeholder tensor 'image_part_input' with dtype float

メインネットワーク入力が正しくないことを参照しているため、これは完全に理解できません。で形状の値をハードコーディングすると、ネットワークが機能しK.zerosます。他の解決策はありますか？

4

1 に答える 1