问题类型

错误

来源

二进制的

tensorflow 版本

tf 2.10，tf-每晚

自定义代码

是的

操作系统平台和分发

Linux操作系统Ubuntu 20.04

移动的设备

• 没有回应 *

Python版本

• 没有回应 *

Bazel版本

• 没有回应 *

GCC/编译器版本

• 没有回应 *

CUDA/cuDNN版本

• 没有回应 *

GPU型号和内存

• 没有回应 *

当前行为？

Applying shuffle+repeat+batch on `tf.data.experimental.CsvDataset` causes check failed and IOT instruction (core dumped) crashes.

重现问题的独立代码

import tensorflow as tf

# The following code is from https://www.tensorflow.org/api_docs/python/tf/data/experimental/CsvDataset

with open('/tmp/my_file0.csv', 'w') as f:
  f.write('abcdefg,4.28E10,5.55E6,12\n')
  f.write('hijklmn,-5.3E14,,2\n')
dataset = tf.data.experimental.CsvDataset(
  "/tmp/my_file0.csv",
  [tf.float32,  # Required field, use dtype or empty tensor
   tf.constant([0.0], dtype=tf.float32),  # Optional field, default to 0.0
   tf.int32,  # Required field, use dtype or empty tensor
  ],
  select_cols=[1,2,3]  # Only parse last three columns
)

# The following operation is causing Check Fail

for e in range(10):
    dataset = dataset.shuffle(1000).repeat().batch(512)

相关日志输出

F tensorflow/core/framework/tensor_shape.cc:404] Check failed: 0 <= new_num_elements (0 vs. -9223372036854775808)
IOT instruction (core dumped)