14_4:変数

TensorFlowを使ってモデルを構築したり訓練するときには、プログラムの実行中に更新したり変更したりできる値をメモリ位置に保持する必要があります。TensorFlowではこれは変数によって行われます。

既存の変数に新しいテンソルを割り当てるにはassign()メソッドを使用します。

const tensor = tf.zeros([3]);
const tvar = tf.variable(tensor); // tvarを初期化
tvar.print();
/*
Tensor
    [0, 0, 0]
*/

// tvar.assign(tf.tensor1d([1, 2, 3, 4, 5]));
// => Error: shape of the new value [5] and previous value [3] must match
tvar.assign(tf.tensor1d([1, 2, 3])); // tvarの値を更新
tvar.print();
/*
Tensor
    [1, 2, 3]
*/

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

CAPTCHA