クロの制作日記

2020-05-07から1日間の記事一覧

tf.functoinとtf.gatherを併用すると勾配が計算できない場合の対処法

最初に Tenosorflow2.0においてtf.functionでデコレートした関数の内部でtf.gatherを使用し、勾配を計算すると以下のようなエラーが出ました。 AssertionError: Expected all args to be Tensors or Variables; but got CompositeTensor エラーの原因 問題の…