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)
相關文章
相關標籤/搜索