【PyTorch】PyTorch中张量(Tensor)计算操作

PyTorch深度学习总结

第五章 PyTorch中张量(Tensor)计算操作


文章目录

  • PyTorch深度学习总结
  • 前言
  • 一、张量比较大小
    • 1、torch.allclose()
    • 2、torch.eq()和torch.equal()
    • 3、ge、gt、le、lt、ne函数
    • 4、torch.isnan()
  • 二、基本运算
    • 1、四则运算(加减乘除)
    • 2、其他计算


前言

上文介绍了PyTorch中张量(Tensor)拆分拼接操作,本文将介绍张量计算操作。


一、张量比较大小

函数 描述
torch.allclose() 比较两个元素是否接近
torch.eq() 逐元素比较是否相等
torch.equal() 判断两个张量是否具有相同的形状和元素
torch.ge() 逐元素比较大于等于
torch.gt() 逐元素比较大于
torch.le() 逐元素比较小于等于
torch.lt() 逐元素比较小于
torch.ne() 逐元素比较不等于
torch.isnan() 判断是否为缺失值

1、torch.allclose()

函数用法:
torch.allclose(A, B, rtol=,atol=)
判断是否接近的公式如下:
∣ A − B ∣ ≤ a t o l + r t o l × ∣ B ∣   . |A-B| \leq atol+rtol\times|B|\,. ABatol+rtol×B.

# 引入库
import torch

# 创建张量A
A = torch.tensor([10.0, 10.0])

# 测试函数
print(torch.allclose(A, A, rtol=0.1, atol=0.01,equal_nan=False))

输出结果为:False


2、torch.eq()和torch.equal()

①函数用法:
torch.eq(A, B)
主要比较元素之间的关系,即两个对应元素是否相等

# 测试函数
print(torch.eq(A, A))

输出结果为:tensor([True, True])


②函数用法:
torch.equal(A, B)
主要比较张量之间的关系,即两个张量形状和大小是否相等

# 测试函数
print(torch.equal(A, A))

输出结果为:True


3、ge、gt、le、lt、ne函数

函数用法:
torch.ge(A, B)
主要用于逐元素比较,看是否大于等于( ≥ \geq )。
注:以上几个函数除本身意义不同外,其他用法几乎相同;故本文只针对torch.ge()进行展示。

# 生成张量
B = torch.tensor([11.0, 9.0])

# 测试函数
print(torch.ge(A, B))

输出结果为:tensor([False, True])


4、torch.isnan()

函数用法:
torch.isnan(A)
判断张量A对应元素是否为缺失值

print(torch.isnan(A))
print(torch.isnan(torch.tensor([0, 1, float("nan")])))

输出结果:
tensor([False, False])
tensor([False, False, True])


二、基本运算

1、四则运算(加减乘除)

生成试验数组:

# 引入库
import torch

# 生成张量
A = torch.arange(6).reshape(2,3)
B = torch.linspace(1, 6, steps=6).reshape(2,3) # 在1-6之间生成5个等步长的元素组成张量
print(A, B)

输出结果为:
tensor([[0, 1, 2], [3, 4, 5]])
tensor([[1., 2., 3.], [4., 5., 6.]])

加减乘除运算为:+-*/
整除://
幂运算为:torch.pow()**

示例:

print(A+B)
print(A-B)
print(A*B)
print(A/B)
print(B//A)
print(A**2)
print(torch.pow(A, 2))

输出结果为:
tensor([[ 1., 3., 5.], [ 7., 9., 11.]])
tensor([[-1., -1., -1.], [-1., -1., -1.]])
tensor([[ 0., 2., 6.], [12., 20., 30.]])
tensor([[0.0000, 0.5000, 0.6667], [0.7500, 0.8000, 0.8333]])
tensor([[inf, 2., 1.], [1., 1., 1.]])
tensor([[ 0, 1, 4], [ 9, 16, 25]])
tensor([[ 0, 1, 4], [ 9, 16, 25]])

2、其他计算

函数 描述
torch.exp() 张量的指数函数
torch.log() 张量的对数函数
torch.sqrt() 张量的平方根
torch.clamp_max() 根据最大值裁剪
torch.clamp_min() 根据最小值裁剪
torch.clamp() 根据范围裁剪
torch.t() 计算矩阵的转置
torch.matmul() 计算矩阵的转置
torch.inverse() 计算矩阵的逆矩阵
torch.trace() 计算矩阵的迹

你可能感兴趣的:(PyTorch,pytorch,人工智能,python)