python代码编写结构

inp_img = input[:, :3]

inp_pose = input[:, 3:3 + pose_dim] if use_input_pose else None

tg_pose_index = 3 + pose_dim if use_input_pose else 6

tg_pose = input[:, tg_pose_index:]

以上为if...else..简写的代码结构。