Tensor又名张量,也是Tensorflow等框架中的重要数据结构。它可以是一个数(标量),一维数组(向量),二维数组或更高维数组。Tensor支持GPU加速。

创建Tensor

几种常见创建Tensor的方法

PyTorch实战福利从入门到精通之二——Tensor

举例:

PyTorch实战福利从入门到精通之二——Tensor

生成全为1的Tensor

PyTorch实战福利从入门到精通之二——Tensor

生成全为0的Tensor

PyTorch实战福利从入门到精通之二——Tensor

随机生成

PyTorch实战福利从入门到精通之二——Tensor

对角线元素为1

PyTorch实战福利从入门到精通之二——Tensor

常用Tensor操作

通过tensor.view方法可以调整tensor的形状,但必须保证调整前后元素总数一致。view不会修改自身的数据,返回的新tensor与源tensor共享内存,也即更改其中的一个,另外一个也会跟着改变。在实际应用中可能经常需要添加或减少某一维度,这时候squeezeunsqueeze两个函数就派上用场了。

PyTorch实战福利从入门到精通之二——Tensor

PyTorch实战福利从入门到精通之二——Tensor

resize是另一种可用来调整size的方法,但与view不同,它可以修改tensor的大小。如果新大小超过了原大小,会自动分配新的内存空间,而如果新大小小于原大小,则之前的数据依旧会被保存,看一个例子。

PyTorch实战福利从入门到精通之二——Tensor

相关文章:

  • 2021-08-31
  • 2021-09-03
  • 2021-06-08
  • 2021-06-28
  • 2022-12-23
  • 2022-01-02
  • 2021-11-23
猜你喜欢
  • 2021-10-26
  • 2021-07-31
  • 2022-01-13
  • 2021-09-22
  • 2021-05-11
  • 2021-08-04
  • 2021-11-07
相关资源
相似解决方案