PyTorch 的 torch.Tensor的常见基本运算

问题：定义两个张量

import torch

tensor1 = torch.tensor([1.0, 2.0, 3.0])
tensor2 = torch.tensor([4.0, 5.0, 6.0])

1、加法

# 1. 加法
result_addition = tensor1 + tensor2
print("加法结果:", result_addition)

加法结果: tensor([5., 7., 9.])

2、减法

# 2. 减法
result_subtraction = tensor1 - tensor2
print("减法结果:", result_subtraction)

减法结果: tensor([-3., -3., -3.])

3、乘法

# 3. 乘法
result_multiplication = tensor1 * tensor2
print("乘法结果:", result_multiplication)

乘法结果: tensor([ 4., 10., 18.])

4、除法

# 4. 除法
result_division = tensor1 / tensor2
print("除法结果:", result_division)

除法结果: tensor([0.2500, 0.4000, 0.5000])

5、点积

# 5. 点积
result_dot_product = torch.dot(tensor1, tensor2)
print("点积结果:", result_dot_product)

点积结果: tensor(32.)

6、幂运算

# 6. 幂运算
exponent = 2
result_power = torch.pow(tensor1, exponent)
print("幂运算结果:", result_power)

幂运算结果: tensor([1., 4., 9.])

7、取开方

# 7. 取开方
result_sqrt = torch.sqrt(tensor1)
print("开方结果:", result_sqrt)

开方结果: tensor([1., 1.4142, 1.7321])

8、取绝对值

# 8. 取绝对值
result_abs = torch.abs(tensor1 - tensor2)
print("取绝对值结果:", result_abs)

取绝对值结果: tensor([3., 3., 3.])

9、更多方法

        参考 PyTorch 文档