深度学习PyTorch-张量简介及创建

 定义:张量是一个数学概念,它在物理学中表示一个物理量在多个方向上的分量。在计算机科学和深度学习中,张量通常被用来表示数据的结构,它是一个多维数组,可以看作是向量和矩阵的高维推广。

使用PyTorch框架创建张量的方法有多种,以下是一些常见的方法:

1.基于Numpy矩阵创建张量:首先需要导入numpy库和torch库。然后,可以使用numpy创建一个矩阵,例如`np.ones((3, 3))`会创建一个3x3的全1矩阵。接着,可以通过`torch.tensor(arr)`将这个numpy矩阵转换为张量,默认情况下这个张量会存放在CPU上。如果想要将其存放在GPU上,可以使用`torch.tensor(arr, device='cuda')`。


2.直接创建张量:可以直接使用`torch.tensor()`根据指定的数据创建一个张量,或者使用`torch.Tensor()`根据指定的形状创建一个张量。此外,还可以创建特定类型的张量,如`torch.IntTensor`、`torch.DoubleTensor`、`torch.FloatTensor`等。


3.创建特定值的张量:可以使用`torch.arange`和`torch.linspace`创建线性张量,使用`torch.randn`创建随机张量,使用`torch.ones`和`torch.ones_like`创建全1张量。


4.创建随机值的张量:如果需要一个随机初始化的张量,可以使用`torch.rand((3, 224, 224))`来创建一个形状为(3, 224, 224)的三维张量。


5.张量的转换:如果已经有一个张量并希望将其从GPU转移到CPU,或者从CPU转移到GPU,可以使用`.cpu()`和`.cuda()`方法进行转换。


6.张量的操作:包括改变张量的形状、获取张量中的元素、拼接和拆分等操作。

要在PyTorch中创建张量,首先需要导入PyTorch库,然后使用`torch.Tensor`或`torch.tensor`函数来创建张量。以下是一个简单的示例:

要在PyTorch中创建张量,首先需要导入PyTorch库,然后使用`torch.Tensor`或`torch.tensor`函数来创建张量。以下是一个简单的示例:

```python
# 导入PyTorch库
import torch

# 创建一个张量
tensor = torch.tensor([1, 2, 3])

# 打印张量
print(tensor)
```

这将创建一个包含三个元素的一维张量。你还可以通过传递一个列表或NumPy数组来创建一个更高维度的张量,例如:

```python
# 创建一个二维张量
two_dim_tensor = torch.tensor([[1, 2], [3, 4]])

# 打印张量
print(two_dim_tensor)
```

这将创建一个2x2的二维张量。 

  • 5
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值