a = torch.tensor([2., 3.], requires_grad=True)
le=torch.tensor([[2., 3.],[1,1]], requires_grad=True)
#print(le+3)
print(le+a)
似乎采取了自动扩充的策略,把0维扩成1维,1维扩成2维的。比如上面的结果,输出:
tensor([[4., 6.],
[3., 4.]], grad_fn=<AddBackward0>)
【 在 feng321 的大作中提到: 】
: import torch
: a = torch.tensor([2., 3.], requires_grad=True)
: b = a + 3
: ...................
--
FROM 120.242.253.*