python - 異なる形状のテンソルをパディングして、2 次元で同じ形状にする方法

Question

形状が (100,43,1024)、(100,37,1024)、(100,42,1024) の 3 つのテンソルがあります。これらすべてのテンソルの 2 番目の次元を 2 番目の次元の最大値、つまりこの場合は 43 にしたいと考えています。パッド機能を使って同じ形にする方法を教えてください。