在PyTorch中進行張量運算非常簡單,可以使用PyTorch提供的各種張量運算函數來實現。下面是一個簡單的例子:
import torch
# 創建兩個張量
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還提供了很多其他函數,如torch.sum()、torch.mean()、torch.max()等,可以用來進行更復雜的運算操作。通過使用這些函數,可以輕松地完成各種張量運算任務。