1、PyTorch使用结尾的下划线表示函数在张量上原地运行(例如Tensor.sqrt_函数)
2、在Tensor中的乘法,其实是对应位置相乘,而不是数学中的矩阵乘法,比如说:
x = torch.ones(2, 2)
y = x + 2
print(y)
z = y*y
print(z)
输出为:
tensor([[3., 3.],
[3., 3.]])
tensor([[9., 9.],
[9., 9.]])
1、PyTorch使用结尾的下划线表示函数在张量上原地运行(例如Tensor.sqrt_函数)
2、在Tensor中的乘法,其实是对应位置相乘,而不是数学中的矩阵乘法,比如说:
x = torch.ones(2, 2)
y = x + 2
print(y)
z = y*y
print(z)
输出为:
tensor([[3., 3.],
[3., 3.]])
tensor([[9., 9.],
[9., 9.]])