pytorch 生成全零tensor 指定type_Tensor与tensor深入分析

最新推荐文章于 2023-07-04 14:26:21 发布

WRovo

最新推荐文章于 2023-07-04 14:26:21 发布

阅读量3.2k

点赞数

文章标签： pytorch 生成全零tensor 指定type

本文介绍了PyTorch中创建全零Tensor与指定type的Tensor的方法。通过两种方式——指定形状大小并初始化及直接赋值数据值，详细解释了torch.tensor与torch.zeros的区别。当输入为5时，torch.tensor创建一个值为5的变量，而torch.zeros创建一个全零矩阵。torch.zeros能指定多维矩阵形状，而torch.tensor更灵活，可直接创建标量。总结来说，torch.zeros适合创建指定形状的全零矩阵，torch.tensor则依据数据值创建变量，包括标量。

摘要由CSDN通过智能技术生成

v2-39bb1573941f44b8950d39672dc4c22e_1440w.jpg?source=172ae18b

在Pytorch官方文档中，对于 `Tensor` 与 `tensor` 是这样定义的：

A torch.Tensor is a multi-dimensional matrix containing elements of a single data type
To create a tensor with pre-existing data, use torch.tensor()

Tensor 是多维矩阵，矩阵的元素都是同一种数据类型。

tensor 需要确切的数据对它进行赋值。

对于变量，创建的方式有两种：创建变量的数据形状大小并初始化；直接赋值确切的数据值

接下来，就讨论 Tensor 与 tensor 创建的特点：

方式一：创建变量的数据形状大小并初始化

指定形状的大小时，会发生这种