如何將幾個張量傳遞給 tensorflow.js 中的 model.predict

Question

我正在嘗試使用 tensorflow.js 來運行我在 python 中創建的模型。 該模型有兩個輸入張量，一個是 [1, 60] 形狀，另一個是 [1, 60, 1]。 我這樣做：

    var array1 = new Array(60).fill(0);
    var tensor1 = tf.tensor(arr1, [1, 60);
    var array2 = new Array(60).fill(1);
    var tensor2 = tf.tensor(arr2, [1, 60, 1]);

    var tensorResult = model.predict([tensor1, tensor2]);

但它給了我這個信息：

錯誤：檢查模型時出錯：您傳遞給模型的張量數組不是模型預期的大小。 預計會看到 2 個張量，但得到了 1 個張量。

模型摘要（）：

__________________________________________________________________________________________________
Layer (type)                    Output shape         Param #     Receives inputs                 
==================================================================================================
l_word (InputLayer)             [null,60]            0                                           
__________________________________________________________________________________________________
l_embWord (Embedding)           [null,60,300]        164395500   l_word[0][0]                    
__________________________________________________________________________________________________
l_company (InputLayer)          [null,60,1]          0                                         
__________________________________________________________________________________________________
l_concat (Concatenate)          [null,60,301]        0           l_embWord[0][0]                 
                                                                 l_company[0][0]                 
__________________________________________________________________________________________________
l_bLstm (Bidirectional)         [null,128]           187392      l_concat[0][0]                  
__________________________________________________________________________________________________
l_dense (Dense)                 [null,16]            2064        l_bLstm[0][0]                   
__________________________________________________________________________________________________
l_dropout (Dropout)             [null,16]            0           l_dense[0][0]                   
__________________________________________________________________________________________________
l_softmax (Dense)               [null,3]             51          l_dropout[0][0]                 
==================================================================================================
Total params: 164585007
Trainable params: 189507
Non-trainable params: 164395500
__________________________________________________________________________________________________

在python中，以下對我來說非常有效：

    model.predict([tensor1, tensor2])

我將不勝感激任何貢獻。

謝謝！

Answer 1

錯誤說明了一切。

您傳遞給模型的張量數組不是模型預期的大小。

作為輸入傳遞給模型的張量的形狀不是模型所期望的