6.5 张量数值计算

6.5.1 基本运算

1）四则运算

• +、-、*、/加减乘除
• add()、sub()、mul()、div()加减乘除，不改变原数据
• add_()、sub_()、mul_()、div_()加减乘除、修改原数据

import torch

tensor1 = torch.randint(1, 9, (2, 3))
print(tensor1)
print(tensor1 + 10)
print()

# add()，不修改原数据
print(tensor1.add(10))
print(tensor1)
print()

# add_()，修改原数据
print(tensor1.add_(10))
print(tensor1)

2）-、neg()、neg_()取负

import torch

tensor1 = torch.tensor([1, 2, 3])
print(-tensor1)
print()

print(tensor1.neg())
print(tensor1)
print()

print(tensor1.neg_())
print(tensor1)

3）**、pow()、pow_()求幂

import torch

tensor1 = torch.tensor([1, 2, 3])
print(tensor1**2)
print()

print(tensor1.pow(2))
print(tensor1)
print()

print(tensor1.pow_(2))
print(tensor1)

4）sqrt()、sqrt_()求平方根

import torch

tensor1 = torch.tensor([1.0, 2.0, 3.0])
print(tensor1.sqrt())
print(tensor1)
print()

print(tensor1.sqrt_())
print(tensor1)

5）exp()、exp_()以e为底数求幂

import torch

tensor1 = torch.tensor([1.0, 2.0, 3.0])
print(2.71828183**tensor1)
print()

print(tensor1.exp())
print(tensor1)
print()

print(tensor1.exp_())
print(tensor1)

6）log()、log_()以e为底求对数

import torch

tensor1 = torch.tensor([1.0, 2.0, 3.0])
print(tensor1.log())
print(tensor1)
print()

print(tensor1.log_())
print(tensor1)