tensor是什么
时间: 2023-07-03 13:20:00 浏览: 53
`tensor` 是 PyTorch 中表示多维数组的基本数据结构。它类似于 Numpy 中的 `ndarray`,但是可以在 GPU 上加速运算。在 PyTorch 中,`tensor` 可以是一个标量、向量、矩阵、甚至是高维数组。`tensor` 中的每个元素都是相同的数据类型,比如 float、int 或 double 等。
`tensor` 可以通过多种方式创建,比如从 Python 列表、Numpy 数组或其他 `tensor` 中创建。在创建 `tensor` 后,可以对它进行各种数学运算、索引和切片操作,也可以将其传递给模型进行前向传播和反向传播计算。
示例代码:
```python
import torch
# 从 Python 列表创建 tensor
x = torch.tensor([1, 2, 3])
print(x)
# 输出:tensor([1, 2, 3])
# 从 Numpy 数组创建 tensor
import numpy as np
arr = np.array([[1, 2], [3, 4]])
y = torch.tensor(arr)
print(y)
# 输出:tensor([[1, 2],
# [3, 4]], dtype=torch.int32)
# 创建一个随机的 2x3 的浮点数 tensor
z = torch.randn(2, 3)
print(z)
# 输出:tensor([[ 0.0726, -1.5313, 0.6149],
# [ 0.3085, -1.0715, -0.2939]])
```
在这个例子中,我们使用了三种方式创建了 `tensor`,分别是从 Python 列表创建、从 Numpy 数组创建和随机创建。我们还打印了每个 `tensor` 的值。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)