Xilinx / DPU-PYNQ

DPU on PYNQ
Apache License 2.0
199 stars 68 forks source link

Why is the Output reshaping nessecary ? #83

Closed HenningCode closed 2 years ago

HenningCode commented 2 years ago

So in the MNIST example Notebook the output gets reshaped with this line.

temp = [j.reshape(1, outputSize) for j in output_data]

Why is this necessary because this Netron Graph of the linked model says the Output should look like 1x10 before the download. The shape is only (1,1,10) , so its not like before the last dense layer and not like the last reshape layer ?

image

Greetings