GAN Custom dataset: cannot reshape array of size

Question

I am training a GAN on a custom dataset titled embeddings with dimensions 50 x 300.

# Custom dataset
class CONCEPTDataset(Dataset):
    def __init__(self, ima