My input tensor is in this shape
[13, 7 , 32]
What I want to delete the middle axis to be
[13, 32]
The squeeze only remove if the dimension is 1.
How to do it by creating relay ops?
Thanks,
Yin
My input tensor is in this shape
[13, 7 , 32]
What I want to delete the middle axis to be
[13, 32]
The squeeze only remove if the dimension is 1.
How to do it by creating relay ops?
Thanks,
Yin