tensorly/tensorly

何を解決するか

TensorLyは、テンソル分解、テンソル代数、テンソル学習のための統一されたPythonライブラリを提供することで、テンソル学習を簡素化します。異なるディープラーニングフレームワーク用に複雑なテンソル操作を再実装する必要がなくなります。

動作方法

高レベルのラッパーとして機能し、柔軟なバックエンドシステムを備えています。デフォルトではNumPyを使用して計算できますが、PyTorch、JAX、TensorFlow、CuPy、Paddleに切り替えることでGPU加速を活用し、スケーラビリティを向上させられます。

対象ユーザー

多次元配列（テンソル）を扱い、さまざまな計算バックエンド上で高度な分解や代数演算を実行したい研究者や開発者向けに設計されています。

特徴

• マルチバックエンド対応: NumPy、PyTorch、TensorFlow、JAX、CuPy、Paddleの間でシームレスに切り替え可能。
• テンソル分解: Tucker分解やCPテンソルなどのメソッドを内蔵サポート。
• ハードウェア加速: 選択したバックエンドに応じてCPUまたはGPUで計算を実行可能。
• テンソル代数: テンソルの作成、展開、折りたたみを扱うツール。