Python対応のtensorFlow またはPytorchで異なる次元のarray掛け算を実現したいです。
例えば,
2D行列(matrix)がN枚があって、要素数N個の実数vector V もあって、
i枚目のmatrixのあらゆる要素にVのi番目の要素にかけたいのです。
そして上記計算をN枚の行列に一括行う関数またはPythonのセンテンスを知りたいのです。 (できれば、GPU対応の同時並列計算的なやり方が望ましい)
broadcastという概念が知っていますが、具体センテンスや関数名は分かりません。
宜しくお願いします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。