给矩阵赋值,变量的值改变,不会改变矩阵的值
import torch
a = torch.zeros((4,5))
print(a)
for i in range(a.size(0)):
temp = torch.ones((5))*i
a[i] = temp
print(a)
给矩阵赋值,变量的值改变,不会改变矩阵的值
import torch
a = torch.zeros((4,5))
print(a)
for i in range(a.size(0)):
temp = torch.ones((5))*i
a[i] = temp
print(a)