我有一组形状为50的(50,128,128,1)图像，在TensorFlow中表示为张量。假设25th和30th图像只是零幅图像，但我不知道哪些图像事先都是零（本例中的25th和30th只是为了使情况更清楚）。我希望删除此类图像，并使用张量为(48,128,128,1)的张量。如何在Tensorflow中实现这一点而又不遍历张量的0th维50次并检查每个图像（如果tf.reduce_sum(tf.abs(image_i))>0）。
我有一组50个形状为（50,128,128,1）的图像，在TensorFlow中表示为张量。假设第25张和第30张图像只是零张图像，但我不知道全部是哪些图像...
您可以使用Dataset.map(some_fn)。在这里，您可以定义some_fn，它将使用逻辑tf.reduce_sum()检查每个张量的值。