outputs_hw1.png975×464 66 KB

ouputs are from final layer and are of shape 1000x10. outputs 2 are pulled 2nd from last layer and are of shape 1000x1024. I want to randomly sample out of my outputs2 to get 1000x10.
I tried this, and it looks like tensors wrapped in a tensor…

output_test4.png932×576 103 KB