javascript - 在tensorflow.js中创建使用null作为批量维度的张量

Question

我的模型需要输入 [null,12]，如错误所示。当我创建张量时

tf.tensor([12 values], [null, 12])

它抛出此错误:张量必须具有由正整数组成的形状，但得到的形状为[,13]。我该如何解决这个问题？

Answer 1

模型需要一个形状为[null, 12]的张量，null表明存在一个正整数的批量维度。 张量本身不能在其维度值之一为空的情况下创建。

可以通过以下方式修复错误

tf.tensor([12 values], [1, 12])

有 12 个值 - 可以看作一批 12 个元素。因此，形状为 [1, 12]。

如果批量大小未知，可以通过这种方式创建张量:

tf.tensor([12 values]).reshape([-1, 12])

在后一种情况下，形状表示:“我不知道有多少个元素，但它是 12 的倍数 - 请自行计算批量大小”。但如果值恰好不是 12 的倍数，则在创建张量时会抛出错误。