DataGenerator(Sequence) -如何检查batch_x和batch_y.shape？

from tensorflow.keras.utils import Sequence
import numpy as np

class MyDataGenerator(Sequence):
    def __init__(self, x_set, y_set, batch_size):
        self.x, self.y = x_set, y_set
        self.batch_size = batch_size

    def __len__(self):
        return int(np.ceil(len(self.x) / float(self.batch_size)))

    def __getitem__(self, idx):
        batch_x = self.x[idx * self.batch_size:(idx + 1) * self.batch_size]
        batch_y = self.y[idx * self.batch_size:(idx + 1) * self.batch_size]

        # 检查batch_x和batch_y的形状
        print(f"Batch {idx} - Input shape: {batch_x.shape}, Label shape: {batch_y.shape}")

        return batch_x, batch_y

# 示例使用
x_train = np.random.rand(1000, 32, 32, 3)  # 假设的输入数据
y_train = np.random.randint(0, 2, (1000, 1))  # 假设的标签数据
batch_size = 32

data_gen = MyDataGenerator(x_train, y_train, batch_size)

# 迭代生成器以查看输出形状
for batch_x, batch_y in data_gen:
    pass  # 这里只是为了展示如何检查形状，实际使用时会在这里进行模型训练