1945年8月15日，日本无条件投降。同年12月，由中、美、英等11国组成的远东国际军事法庭决定对发动侵略战争的日本军国主义分子进行审判。中国国民政府接到驻日盟军统帅总部的通知，便指示外交部和司法部遴选派往东京的外交、司法人员。两部官员立刻从全国范围内挑选精英，组成了具有强大实力的参审班子，44岁的武汉大学法律系教授吴学义被选派担任检察官顾问。
1946年5月，吴学义随代表团从上海乘专机飞赴日本东京。5月3日，正式开庭的前一天，各国法官、检察官、顾问、律师以及有关人员召开一次预备会议。11个对日参战国的法官都庄严地坐在审判席上，中国法官梅汝璈身穿崭新的法官服，威严地坐在审判席的中间，审判席后面插着参战国的国旗，美国插在第一位，中国插在第二位。吴学义顾问一看中国国旗插在第二位，立刻向梅法官打手势，伸出了右手食指，意思是中国国旗应插在第一位，梅见吴的手势后，心领神会，立即向庭长韦伯提出：“中国国旗应插在第一位。”并与美国法官进行激烈的争论，唇枪舌剑，终于让中国国旗就此插在了第一位，美国国旗则移至第二位。这是自1840年鸦片战争之后，中国代表团出席国际会议有史以来国旗插在首位的第一次！国内新闻媒介立即发了这一重大新闻，有的报纸还及时刊出“号外”。
除此之外，吴学义还为东京审判作了大量幕后工作。日本投降后有计划地销毁了大量侵略证据，为搜集证据，中国检察官向盟军总部提出申请要求查阅日本政府的机密档案，得到许可后，吴学义几乎每天都到被查封的日本行政当局档案库，从字里行间发掘日本战犯的罪行证据。面对成千上万的档案，吴学义与同行刘子健，两个人就在那里找，找了一个礼拜。
1946年5月，吴学义随代表团从上海乘专机飞赴日本东京。5月3日，正式开庭的前一天，各国法官、检察官、顾问、律师以及有关人员召开一次预备会议。11个对日参战国的法官都庄严地坐在审判席上，中国法官梅汝璈身穿崭新的法官服，威严地坐在审判席的中间，审判席后面插着参战国的国旗，美国插在第一位，中国插在第二位。吴学义顾问一看中国国旗插在第二位，立刻向梅法官打手势，伸出了右手食指，意思是中国国旗应插在第一位，梅见吴的手势后，心领神会，立即向庭长韦伯提出：“中国国旗应插在第一位。”并与美国法官进行激烈的争论，唇枪舌剑，终于让中国国旗就此插在了第一位，美国国旗则移至第二位。这是自1840年鸦片战争之后，中国代表团出席国际会议有史以来国旗插在首位的第一次！国内新闻媒介立即发了这一重大新闻，有的报纸还及时刊出“号外”。
除此之外，吴学义还为东京审判作了大量幕后工作。日本投降后有计划地销毁了大量侵略证据，为搜集证据，中国检察官向盟军总部提出申请要求查阅日本政府的机密档案，得到许可后，吴学义几乎每天都到被查封的日本行政当局档案库，从字里行间发掘日本战犯的罪行证据。面对成千上万的档案，吴学义与同行刘子健，两个人就在那里找，找了一个礼拜。
