Условия тензорного среза для присваивания - PullRequest
0 голосов
/ 26 мая 2020

Почему этот '' 'tensor_ques [1: 3,1] = 0' '' работает, а этот '' 'tensor_ques [1: 3,1] = [0,0]' '' не работает tensor_ques is a torch.tensor ()

Я знаю, что [0,0] не тензор, тогда почему = 0 работает нормально

1 Ответ

0 голосов
/ 26 мая 2020

Это не работает, потому что [0,0] - это список, а torch не позволяет вам назначать список тензору. Это будет работать только в том случае, если назначаемый элемент также является тензором, т.е. tensor_ques[1:3,1] = torch.tensor([0,0]).

...