最適化関数によるテンソルの更新（２）

optimの提供する最適化器には、複数のテンソルを与えることができます。

ここでは２つのテンソルを与える例を示します。(1) のように、リストで渡しています。

code:sgd.py

import torch as pt

import torch.optim as optim

lr = 1 # 学習率

print('# 実験1')

print('* Before')

print(x1)

print(x1.grad)

print(x2)

print(x2.grad)

optimizer.step() # ここで操作

print('* After')

print(x1)

print(x2)

(1)で最適化器に渡したx1, x2に対して操作を行うことができます。

このように、optimの提供する最適化器は複数のテンソルをターゲットにすることが可能です。

/icons/hr.icon

※ ブラウザのバックボタンで戻る