tensorflow 将3DTensor数组转换为单个4DTensor

vlf7wbxs  于 2023-02-16  发布在  其他
关注(0)|答案(1)|浏览(196)

我试图将一个3DTensor(图像)数组转换为一个4DTensor,这样我就可以将它们作为值传递给model.fit,而Tensor3D似乎不接受Tensor3D数组。
我们的想法是

4dTensor = tf.tensor4d(batch)

我实际上使用的是javascript,但python或js解决方案可能会工作,因为Tensorflow API是类似的。
此过程的错误是:

Argument of type 'Tensor4D' is not assignable to parameter of type 'Tensor3D[]'.
  Type 'Tensor<Rank.R4>' is missing the following properties from type 'Tensor3D[]': length, pop, push, join, and 26 more.ts(2345)
pw136qt2

pw136qt21#

您可能需要使用tf.stack()

4dTensor = tf.stack(batch)

相关问题