预期density_Dense1_input的形状为[null,2],但数组的形状为[1,1]

时间:2019-03-29 12:41:20

标签: tensorflow tensorflow.js

我知道这个问题已经被问过很多遍了,但是我尝试了大多数,但仍然无法弄清楚我错了。

在tensorflow.js中,我有一个模型,可以在其中添加输入层

this.model.add(this.tf.layers.dense({ units: 10, inputShape: 2, activation: 'relu' }));

我尝试将以此方式构建的数组作为输入传递

myArray = [[1, 0.5], [1, 0.3], [1,0.7]]

并转换为张量

myTensor =  this.tf.tensor(myArray);

如果我尝试获得myTensor的形状,我会得到

[8, 2]

无论如何,当我开始训练模型时,我会立即出错

Error: Error when checking : expected dense_Dense1_input to have shape [null,2] but got array with shape [1,1].

如何为myArray建模以便能够在其上训练我的模型?

0 个答案:

没有答案