Запись для градиента в PyTorch
У обычного тензора атрибут
requires_grad по умолчанию
выключен. Создадим целое число
и выведем этот флаг:
import torch
count = torch.tensor(3)
print(count.requires_grad) # выведет False
Для дробного значения запись включают
при создании: у атрибута
requires_grad нужно
значение Истина. Тогда флаг
записи включён:
import torch
weight = torch.tensor(2.5, requires_grad=True)
print(weight.requires_grad) # выведет True
Соберите одно дробное число
1.0 так, чтобы для него
велась запись операций, и выведите
значение флага записи.
Соберите ряд из целых 2,
4 и 6 и выведите
флаг записи у этого ряда.
Соберите одно дробное число
0.5 с записью и ряд
из двух целых 1 и 2
без неё. Выведите оба флага
записи.