Input node retains batch dimension

neuromorphs / NIRTorch

PyTorch helper module to translate to and from NIR

https://neuroir.org/docs

BSD 3-Clause "New" or "Revised" License

7 stars 3 forks source link

Input node retains batch dimension #17

Open sheiksadique opened 1 year ago

sheiksadique commented 1 year ago

Since in pytorch, the data is always expected to have the batch dimension, this shape is also retained in the input shape. This is not consistent with NIR specifications that does not include the batch dimension.