@chenchr: Thanks for your suggestion. I tried unsqueeze. But I could not get the desired answer. Will it be possible for you to show me the code, how to get the final result I’m getting using numpy?
Thank you!