-
Notifications
You must be signed in to change notification settings - Fork 86
Open
Description
I have been reading thorugh the code for the function "MaxUnpoolWithArgmax" since I'm interested in the logic to extract the indexes [b,y,x,c] from the argmax tensor. I cannot understand the maths behind to extract the index "y" and "x". Could you elaborate a bit on how did you get these expressions? :
-
y = argmax // (output_shape[2] * output_shape[3])
-
x = argmax % (output_shape[2] * output_shape[3]) // output_shape[3]
Thank you so much
Pedro
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels