HeyTensor

HeyTensorは、PyTorchで発生するRuntimeErrorやテンソル形状の不一致を素早く診断・修正できるAI支援デバッグツールです。Conv2d・Linear・LSTMをはじめとする14種類以上のレイヤーに対応しており、各レイヤーの入出力テンソル形状を正確に計算する機能を備えています。

ディープラーニングモデルの開発・実験を行うエンジニアやリサーチャーを主な対象としており、PyTorchのエラーメッセージに日々悩まされている方に特に適しています。

モデル構築中に「RuntimeError: mat1 and mat2 shapes cannot be multiplied」などのエラーが出た際に、エラー文をそのままペーストするだけで修正案を即座に取得できます。テンソル形状の設計段階でも活用でき、試行錯誤にかかる時間を大幅に短縮できます。