The shape is 32, 512. However it will be regardless the number of rows in the table. This is the main problem that i have