Wpis z mikrobloga

Mam zagwozdkę. Moj model jest w postaci wymiaru [??,1]. Po treningu i podzieleniu datasetu na batch o rozmiarze 335, chciałbym zmienić rozmiar batcha na pełny rozmiar datasetu lecz wtedy moj dataset jest w postaci [??,335,1], tak jakby robił batch'e z uprzednio utworzonych batch'y. Nie mogę znaleźć metody resetującej dataset do pierwotnej postaci i usuwającej utworzone batch'e. Miał ktoś z was taki problem?
#tensorflow #python