俺の頭がポンコツすぎて、よくどの変数にどんなシェイプのテンソルを突っ込んだかを忘れちゃうんだけどさ
静的にできる範囲でテンソルのシェイプを推論して、VSCode上で表示してくれるエクステンションとかあったら教えてくれんかな?