小编ROS*_*ROS的帖子

我在用

file_data = tf.data.Dataset.list_files("../*.png")

收集图像文件以在 TensorFlow 中进行训练，但想访问收集的文件名列表，以便我可以执行标签查找。

调用 sess.run([file_data]) 已经失败：

TypeError: Fetch argument <TensorSliceDataset shapes: (), types: tf.string> has invalid type <class 'tensorflow.python.data.ops.dataset_ops.TensorSliceDataset'>, must be a string or Tensor. (Can not convert a TensorSliceDataset into a Tensor or Operation.)

我可以使用其他任何方法吗？

我正在使用 PyTorch 版本 1.5.0。当我传递大小为 [8,21,400,400] 的输入火炬张量和大小为 [8,400,400] 的目标时，程序会引发 TypeError：对 0 维张量进行迭代。然而，参数的维度分别为 4 和 3。

什么可能导致此错误？

回溯指向 torch\tensor.py 的iter函数。

Traceback (most recent call last):
  File "train.py", line 108, in <module>
    loss, accuracy = lossLayer(pred2, targetBatch)
  File "C:\Users\PC\anaconda3\lib\site-packages\torch\tensor.py", line 462, in __iter__
    raise TypeError('iteration over a 0-d tensor')
TypeError: iteration over a 0-d tensor