如何解决张量等级不同的问题

Question

我有一个tensorflow.js模型，我为我的模型创建了伪输入，它是：

a=tf.tensor2d([1,2,3,4,5,6,7,8],[8,1],dtype="int32")

我已经使用以下行将其输入到我的模型中：

model.then(m => m.executeAsync({"input_ids":a,"attention_mask":a,"token_type_ids":a}))

其中三个输入具有相同的值，但我收到以下错误消息：

Uncaught (in promise) Error: Error in matMul: inputs must have the same rank of at least 2, got ranks 3 and 2

有人知道我做错了什么，导致我的输入中有排名3的张量吗？非常感谢。

Answer 1

您可以做的事情：

1. 检查模型的输入尺寸
2. 通过模型将输入张量维度重塑为预期维度
3. 然后推断