pytorch add_

# a = a + 4 * 5

import torch
a = torch.tensor([1,2,3])

a.data.add_(torch.tensor(4),torch.tensor(5))
print(a) # tensor([21, 22, 23])
©️2020 CSDN 皮肤主题: 创作都市 设计师:CSDN官方博客 返回首页