It does not take in input of 1 and 0,
I want to create a tensor of the same dimensions as the original tensor target.
I can only create one less dimension tensor using torch.where. How can I also get the 4 dimension?
# Wanted object_temp Shape -> ( 24, 7, 7, 30)
#target Shape -> ( 24, 7, 7, 30)