Transpose do not change output dformat
-
aidge_core
: 0.5.1
Problem description
When forwardDims Transpose.cpp, the output DataFormat is automatically set to DataFormat::Dafault. However, if our input is in NCHW and we apply transposition [0,2,3,1], the output should be in NHWC. I suggest adding a function to DataFormat.hpp
DataFormat getTransposeOutputDataFormat( const DataFormat inputDataFormat, const std::vector<DimSize_t> outputDimsOrder)
And use it to set the correct output DataFormat during the forwardDims() of Transpose.
TODO
- Method in DataFormat.hpp
- Change Transpose::forwardDims
- Add tests