Can not squeeze dim[1], expected a dimension of 1, got 21

Это связано с тем, что документация по tf требует, чтобы метка была единственным значением, а не значением с горячим кодированием, таким как [1,0,0], [0,1,0], [0,0,1].