tensorflow を使用して、完全な計算グラフ (前方、後方、およびパラメーターの更新を含む) を取得したいと考えています。私は tf.functions を試しましたが、得られたグラフは細かすぎます。多くの演算 (Adam など) が小さな演算子 (add、mul、div など) に分割されているためです。粗粒度の操作レベルのグラフを取得できる方法はありますか? どうもありがとう！
質問する
31 次
tensorflow を使用して、完全な計算グラフ (前方、後方、およびパラメーターの更新を含む) を取得したいと考えています。私は tf.functions を試しましたが、得られたグラフは細かすぎます。多くの演算 (Adam など) が小さな演算子 (add、mul、div など) に分割されているためです。粗粒度の操作レベルのグラフを取得できる方法はありますか? どうもありがとう！