pytorch创建tensor
时间: 2023-10-10 14:12:47 浏览: 94
PyTorch是一个开源的机器学习框架,可以用于创建、训练和部署深度学习模型。在PyTorch中,我们可以使用`torch.Tensor`类来创建张量(tensor),它是PyTorch中最基本的数据结构。
以下是一些常见的创建tensor的方法:
1. 从Python列表或NumPy数组创建tensor:
```python
import torch
import numpy as np
# 从Python列表创建tensor
x = torch.tensor([1, 2, 3, 4])
print(x)
# 从NumPy数组创建tensor
y = torch.from_numpy(np.array([5, 6, 7, 8]))
print(y)
```
2. 创建全0或全1的tensor:
```python
# 创建全0的tensor
a = torch.zeros(3, 4)
print(a)
# 创建全1的tensor
b = torch.ones(2, 3, 4)
print(b)
```
3. 创建随机数tensor:
```python
# 创建随机数tensor(均匀分布)
c = torch.rand(2, 3)
print(c)
# 创建随机数tensor(标准正态分布)
d = torch.randn(3, 4)
print(d)
```
4. 创建指定形状的tensor:
```python
# 创建指定形状的tensor
e = torch.empty(2, 3)
print(e)
# 创建指定形状的tensor(默认dtype为float32)
f = torch.zeros((2, 3), dtype=torch.int)
print(f)
```
阅读全文