torch创建一个张量
时间: 2024-07-27 18:00:53 浏览: 97
PyTorch中张量的创建方法
在PyTorch中,你可以使用`torch.Tensor`类或者相关的初始化函数来创建一个张量。下面是一些基本的方法:
1. **直接创建**:
```python
import torch
# 创建一个全零张量
tensor_0 = torch.tensor([0., 0., 0.])
# 创建一个随机张量
tensor_random = torch.randn(2, 3)
```
2. **从numpy数组转换**:
```python
import numpy as np
numpy_array = np.array([[1, 2], [3, 4]])
tensor_from_numpy = torch.from_numpy(numpy_array)
```
3. **固定形状和数据类型**:
```python
shape = (5, 7) # 张量的维度
dtype = torch.float32 # 数据类型,如float、int等
tensor_fixed_shape = torch.empty(shape, dtype=dtype)
```
4. **创建特定维度的填充张量**:
```python
tensor_filled = torch.zeros((3, 4), dtype=torch.long, fill_value=99)
```
阅读全文