output_size, scope=scope)# Each arg is batch size x arg dim. Add dimensions to enable broadcasting.fori, arginenumerate(args):forjinrange(len(args)):ifj == i:continueargs[i