カテゴリー: TensorFlow

TensorFlowでもラプラシアンピラミッドを作る

796{icon} {views}以前作ったPyTorchのラプラシアンピラミッドをTensorFlow2.0に移植しました。何かと便利なラプラシアンピラミッドをつかってみよう。 環境 TensorFlow2.0 CPU […]...

TensorFlow2.0のGradientTapeを複数使う場合のサンプル

4.5k{icon} {views} TF2.0系で少し複雑なモデルを訓練するときに、GradientTapeを複数使うことがあります。例として、微分を取りたい場所が2箇所あるケースや、2階微分を取りたいケースを挙げます […]...

tf.tensordotで行列積を表現するための設定

4.1k{icon} {views} TensorFlowのtensordotという関数はとても強力で、テンソルに対する行列積に対する計算をだいたい表現できます。しかし、軸の設定がいまいちよくわからなかったので、確かめて […]...