# Codeblock 10 print(out[0].shape) # branch0 print(out[1].shape) # branch1 print(out[2].shape) # x