在tensorflow.js中以零作为批次维度创建张量

时间:2020-04-09 06:15:43

标签: javascript tensorflow tensorflow.js

我的模型期望错误提示输入[null,12]。当我创建张量时

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

它抛出此Error: Tensor must have a shape comprised of positive integers but got shape [,13]。我该如何解决?

1 个答案:

答案 0 :(得分:1)

模型期望形状为[null, 12]的张量,null表示存在一个正整数的批次尺寸。 张量本身不能使用其维度值之一为null来创建。

错误可以通过

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

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

如果批次大小未知,则可以通过以下方式创建张量:

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

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