keras 一维张量上 Tensorflow 迭代 = None

xe55xuns 于 2022-11-13 发布在其他

关注(0)|答案(1)|浏览(149)

我正在处理tensorflow中的一个自定义keras图层，我需要迭代一个Tensor。问题是Tensor有形状（无，...，...，...）。我不知道如何迭代。我尝试过

for i in range(len(x[:, 0, 0, 0])):

甚至是tf.map_fn，但是这两个函数都要迭代无限次，如何在一个一维为零的Tensor上迭代呢？

1条答案

我认为添加更多的信息会有所帮助。什么是Tensor？你的训练数据？tf.map应该仍然有效，但可能需要首先禁用渴望执行，如下所示：

import tensorflow as tf
tf.compat.v1.disable_eager_execution()