Hi,
I guess I’m failing to see how to extend this to the n-dimensional case. @richard
Assume I have a tensor where the first two dims are batch and channel, and the last three correspond to xyz space:
A = torch.randn(b,c,32,32,32)
What I would like to do is to binarize along the x dimension (dim=2) for any batch or channel, i.e for every yz location I want to set the maximum value along x-axis to 1 and the rest to zero. Is there a way of doing this?
Thank you!