溫馨提示×

PyTorch中怎么進行張量運算

小億
86
2024-05-10 15:30:52
欄目: 深度學習

在PyTorch中進行張量運算非常簡單，可以使用PyTorch提供的各種張量運算函數(shù)來實現(xiàn)。下面是一個簡單的例子：

import torch

# 創(chuàng)建兩個張量
x = torch.tensor([1.0, 2.0, 3.0])
y = torch.tensor([4.0, 5.0, 6.0])

# 加法運算
z = x + y
print(z)

# 乘法運算
z = x * y
print(z)

# 矩陣乘法
x = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
y = torch.tensor([[5.0, 6.0], [7.0, 8.0]])
z = torch.matmul(x, y)
print(z)

除了加法、乘法等基本的張量運算外，PyTorch還提供了很多其他函數(shù)，如torch.sum()、torch.mean()、torch.max()等，可以用來進行更復雜的運算操作。通過使用這些函數(shù)，可以輕松地完成各種張量運算任務(wù)。

0 踩