【深度学习】初识ndarray

文章目录

前言

🌴 主要介绍pytorch中对于ndarray的一些基础操作;

  1. 矩阵操作

1.1 ndarray

🌴 ndarray(N-dimensional array)意思是n维数组;

🌴 n维数组,也称为张量(tensor)或矩阵(matrix);

🌴 实际上三者略有不同,我们暂时不区分,在本文中数组、张量和矩阵为同义词;

🌴 一个零维数组,也就是一个点,我们称作标量(scalar);一个一维数组,也就是一行或者一列,我们称作向量(vector);一个二维的数组,我们称作矩阵(matrix);

1.2 创建行向量

🌴 代码如下:

x = torch.arange(12)
x

🌴 输出结果为:

【深度学习】初识ndarray

1.3 改变张量的形状

🌴 张量的形状可以改变,比如把一个1 X 12的张量改为一个3 X 4的张量;

🌴 代码如下:

X = x.reshape(3, 4)
X

🌴 输出结果为:

【深度学习】初识ndarray

🌴 对于二维张量来说,如果知道张量的元素个数,行数和列数只需知道一个,我们就可以确定另一个,所以我们可以只输入行数或者列数,另一个用-1代替;

【深度学习】初识ndarray
【深度学习】初识ndarray

1.4 获取张量中的元素个数

🌴 获取张量中的元素个数,我们使用numel函数(number of elements),代码如下:

x.numel()

🌴 输出结果为:

【深度学习】初识ndarray
  1. 创建矩阵

2.1 创建一个全是0的矩阵

🌴 代码如下:

torch.zeros((2, 3, 4))

🌴 输出结果为:

【深度学习】初识ndarray

2.2 创建一个全是1的矩阵

🌴 代码如下:

torch.ones((2, 3, 4))

🌴 输出结果为:

【深度学习】初识ndarray

2.3 创建随机数矩阵

🌴 代码如下:

torch.randn(2 ,3, 4)

🌴 输出结果为:

【深度学习】初识ndarray
  1. 矩阵运算

3.1 矩阵的加减乘除运算

🌴 我们可以对 两个相同形状的矩阵进行加减乘除运算;

🌴 代码如下:

x = torch.tensor([1.0, 2, 4, 8])
y = torch.tensor([2, 2, 2, 2])
x + y, x - y, x * y, x / y, x ** y

🌴 输出结果为:

【深度学习】初识ndarray

3.2 矩阵元素类型转换

🌴 代码如下:

X = torch.arange(12, dtype=torch.float32)
X

🌴 输出结果为:

【深度学习】初识ndarray

3.3 矩阵连接

🌴 代码如下:

X = torch.arange(12, dtype=torch.float32).reshape((3,4))
Y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])
torch.cat((X, Y), dim=0), torch.cat((X, Y), dim=1)

🌴 输出结果为:

【深度学习】初识ndarray
  1. 矩阵的索引

4.1 矩阵的标量索引

🌴 矩阵的索引就是元素下标,python对于矩阵中标量的索引与C语言相同;

🌴 代码如下:

X[1,3]

🌴输出结果为:

【深度学习】初识ndarray

4.2 矩阵的向量索引

🌴 与标量索引不同的是,python对于向量元素下标的标记方式是把”,”换成”:”,采用的是左闭右开区间,如x[1:3]表示矩阵x的第二行第三行的元素,用数学表达式表示也就是[1,3);

🌴 代码入下:

X[1:3]

🌴 输出结果为:

【深度学习】初识ndarray

🌴 访问某一列向量时,使用矩阵名称[:,n],n为列的下标;
🌴 代码如下:

X[:,1]

🌴 输出结果为:

【深度学习】初识ndarray

🌴 如果不写逗号右边的数,会默认为输出整个矩阵;

【深度学习】初识ndarray

4.3 更加精确的索引

🌴 若想要一个更加精确的索引区域,我们还可以在逗号两边的”:”上做文章;
🌴我们已经知道,X[:,:]输出的是整个矩阵X,那么如果我们输入 下列代码会有什么不同呢?

X[1:,:],X[:1,:],X[:,1:],X[:,:1]

🌴 输出结果为:

【深度学习】初识ndarray

🌴 我们可以画图对比一下四种索引的区别;

【深度学习】初识ndarray

🌴 X[:,:]表示整个矩阵,我们可以在逗号左边添加数字对行进行操作;在逗号左边的冒号左边添加数字”1″,表示返回范围为[1,n+1]的矩阵(n为最后一行的下标);

🌴 在逗号左边的冒号的右边添加数字”1″,表示返回范围为[0,1)的矩阵;

🌴 在逗号右边添加数字对列进行操作;在逗号的右边的冒号的左边添加数字”1″,表示返回范围为[1,n+1]的矩阵(n为最后一列的下标);

🌴 在逗号的右边冒号的右边添加数字”1″,表示返回范围为[0,1)的矩阵;

🌴 我们也可以在逗号的左右两边的冒号的左右两边都添加数字,比如X[1:2,1:3];

【深度学习】初识ndarray

🌴 逗号左边的[1,2]表示的行下标的范围是[1,2);逗号右边表示的列下标的范围是[1,3),取它们的交集,所以为tensor[5,6],这就是更精确的数据处理;

  1. 内存节省

🌴 对于变量X,Y,进行”Y=X+Y”或”Y+=X”运算时,python与C的逻辑并不同;

🌴 在python中,进行”Y=X+Y”或”Y+=X”运算后,会增加一块新的内存”Y”,这块内存与原来的”Y”的地址不同,将”Y+X”的值赋给新的变量”Y”后,旧的”Y”的内存地址将被取消引用,转给引用新的”Y”的地址;

🌴 我们可以写一段代码来验证一下:

before = id(Y)
Y=Y+X
id(Y) == before
#若为true,证明内存地址没有改变,否则,证明内存地址改变;

🌴 输出结果为:

【深度学习】初识ndarray

🌴 在C语言中,系统会为每个变量分配内存空间,当改变变量的值时,改变的是内存空间中的值,变量的地址是不改变的。

🌴 我们同样可以使用代码来验证一下:

#include
int main() {
    int a = 1;
    int b = 2;
    printf("求和前a的地址为:%p\n", &a);
    a = a + b;
    printf("求和后a的地址为:%p\n", &a);
    return 0;
}

🌴 输出结果为:

【深度学习】初识ndarray

🌴 这就意味着,当我们使用这种方式计算,就要不断地开辟新的空间,一是容易造成内存浪费;二是可能仍有部分参数在引用旧的”Y”的地址,导致程序运行错误;

🌴 那么如何解决这个问题呢?我们可以使用Y[:]=X+Y,就跟C中一样了;

🌴 可以用代码再次验证一下:

before = id(Y)
Y[:]=Y+X
id(Y) == before

🌴 输出结果为:

【深度学习】初识ndarray

总结

如果你认为它有用,你不妨喜欢它。

[En]

If you think it’s useful, you might as well like it.

Original: https://blog.csdn.net/m0_60266328/article/details/124567859
Author: 马孔多镇长
Title: 【深度学习】初识ndarray

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/497422/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球