-
|
I am trying to remove extra dimension from the mentioned tensor. But squeeze() is not working. For your kind information my current device ion "cpu". So can anyone tell me why it is not working?
|
Beta Was this translation helpful? Give feedback.
Replies: 2 comments 2 replies
-
|
squeeze only removes 1 dimension shapes. You shape is 5,4 - there is no 1 size dim to be removed
|
Beta Was this translation helpful? Give feedback.
-
|
Yes, now I understood. Thank you so much. |
Beta Was this translation helpful? Give feedback.
-
|
@ryan-popa is correct, thank you for sharing/helping out! |
Beta Was this translation helpful? Give feedback.
-
|
Thank you Rayn Popa. Now its clear. |
Beta Was this translation helpful? Give feedback.
squeeze only removes 1 dimension shapes. You shape is 5,4 - there is no 1 size dim to be removed