主题

Pytorch 学习

对于 tensor 的了解

我认为 tensor 是一种数据的表现形式，经过底层封装，便于在某种实现架构内实现快速运算。

导入

import torch

建立

• 建立一个空的 tensor，不初始化

• 建立一个全 0 的 tensor

• 建立一个全 1 的 tensor

• 建立一个随机的 tensor

• 指定 tensor 的数据类型

基础操作

• 加减运算：两个 tensor 的维度应该一致，直接使用 + 和 - 操作符即可。

• 按位乘和按位除

• 按位求绝对值

• 矩阵运算

• 查看 tensor 的 size

• 改变 tensor 的尺寸，要求改变后总数据量不变，如 2 4 = 1 8；-1 表示待计算，由总数据量和其他维度计算得到

• 获取 tensor 中某个位置的元素