好像找到了。
因为 TiDB 的 id 不是连续的，所以 $returningId() 的结果是一个巨大的数组，从 1 开始直到一个奇怪的数字（好像也不是最近插入的 id），于是就超了内存。没有办法解决，只能分开，插入一次查询一次。
$returningId()