pytorch总结学习系列-操作

算术操作 在PyTorch中,同一种操作可能有很多种形式,下⾯用加法作为例⼦。 加法形式⼀ x = torch.tensor([5.5, 3]) y = torch.rand(5, 3) print(x + y) 加法形式⼆ print(torch.add(x, y))  还可指定输出: result = torch.empty(5, 3) torch.add(x, y, out=result)
相关文章
相关标签/搜索