Pytorch tensor Quickstart

Pytorch tensor

0. 环境介绍

环境使用 Kaggle 里免费建立的 Notebook

教程使用李沐老师的 动手学深度学习 网站和 视频讲解

引入 pytorch

import torch

1. tensor

1.1 基本操作

张量（tensor）数值组成的数组，数组可以有多个维度

1.1.1 arange() 生成向量

1.1.2 shape 输出张量形状

1.1.3 numel() 返回张量中元素总数

1.1.4 reshape() 改变一个张量的形状

reshape() 适用于元素数量相同的所有张量

1.1.5 zeros() 全 0 阵

1.1.6 ones() 全 1 阵

1.1.7 tensor 列表赋值

1.2 tensor 运算

1.2.1 常见的标准算术运算符

常见的标准算术运算符（+，-，*，/，**）都是按照对应元素运算，** 是求幂运算

1.2.2 exp() 指数运算

1.2.3 cat() 连接张量，dim 参数值默认为 0（下面拼接），为 1 时表示右边拼接

1.2.4 == 逻辑运算

1.2.5 sum() 对张量内元素求和求和

1.2.6 广播机制 元素操作，b 张量将 0，1 分别进行列扩展再对应元素运算

1.2.7 张量访问

1.3 执行原地操作

运行一些操作可能会导致新结果分配内存

执行原地操作不会分配新的内存，zeros_like 表示生成一个与 Y 张量 shape 相同的全 0 阵

1.4 转换为 Numpy 张量

1.5 1*1 张量转换为标量