скопировать конструкцию из тензора: ПРЕДУПРЕЖДЕНИЕ ПОЛЬЗОВАТЕЛЯ

Question

Я создаю случайный тензор из нормального распределения, и так как этот тензор служит весом в NN, для добавления атрибутов require_grad я использую torch.tensor (), как показано ниже:

import torch 

input_dim, hidden_dim = 3, 5

norm = torch.distributions.normal.Normal(loc=0, scale=0.01)
W = norm.sample((input_dim, hidden_dim))
W = torch.tensor(W, requires_grad=True)

Iя получаю предупреждение об ошибке, как показано ниже:

    UserWarning: To copy construct from a tensor, 
    it is recommended to use sourceTensor.clone().detach() or 
sourceTensor.clone().detach().requires_grad_(True), 
rather than torch.tensor(sourceTensor).

Есть ли альтернативный способ достижения вышеуказанного? Спасибо

Answer 1

Вы можете просто установить W.requires_grad на True

import torch 

input_dim, hidden_dim = 3, 5

norm = torch.distributions.normal.Normal(loc=0, scale=0.01)
W = norm.sample((input_dim, hidden_dim))
W.requires_grad = True