pytorth 多维tensor的操作
时间: 2024-04-21 08:22:53 浏览: 116
PyTorch提供了丰富的多维张量操作。以下是一些常用的操作示例:
1. 创建张量:
```python
import torch
# 创建一个2x3的浮点数张量
tensor = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
# 创建一个3x3的零张量
zeros = torch.zeros((3, 3))
# 创建一个5x5的随机整数张量(取值范围为[0, 10))
random_ints = torch.randint(0, 10, (5, 5))
```
2. 张量的基本操作:
```python
# 获取张量的形状
shape = tensor.shape
# 获取张量的维度数
dim = tensor.dim()
# 获取张量的元素数量
num_elements = tensor.numel()
# 访问单个元素
element = tensor[0, 1] # 访问第一行、第二列的元素
# 修改单个元素
tensor[1, 2] = 7.0
# 张量的切片操作
sub_tensor = tensor[:, 1:3] # 获取所有行的第二到第三列的子张量
```
3. 张量的数学运算:
```python
import torch
# 加法
result = tensor1 + tensor2
# 减法
result = tensor1 - tensor2
# 乘法(逐元素相乘)
result = tensor1 * tensor2
# 矩阵乘法
result = torch.matmul(tensor1, tensor2)
# 平均值
mean = tensor.mean()
# 标准差
std = tensor.std()
# 求和
sum = tensor.sum()
# 最大值
max_value = tensor.max()
# 最小值
min_value = tensor.min()
```
这只是一些基本的操作示例,PyTorch还提供了许多其他功能丰富的操作和函数,可根据具体需求进行深入学习和使用。
阅读全文