Pytorch基本语法

本文介绍了PyTorch中的张量操作,包括`torch.is_tensor`和`torch.is_storage`用于检查对象是否为张量或存储对象,`torch.numel`计算张量元素数量,以及`torch.linspace`、`torch.ones`、`torch.rand`和`torch.randn`用于生成不同类型的张量。这些函数在构建和操作神经网络中至关重要。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

包 torch 包含了多维张量的数据结构以及基于其上的多种数学操作。另外,它也提供了多种工具,其中一些可以更有效地对张量和任意类型进行序列化。 

 张量 Tensors

  1. torch.is_tensor
    obj=torch.full((2,3),6)
    print(obj)
    print(torch.is_tensor(obj))

    如果obj 是一个pytorch张量,则返回True

  2. torch.is_storage

    obj=torch.full((2,3),6)
    print(obj)
    print(torch.is_storage(obj))

    如何obj 是一个pytorch storage对象,则返回True

  3. torch.numel

    a = torch.randn(1,2,3,4,5)
    print(torch.numel(a))
    
    a = torch.zeros(4,4)
    print(torch.numel(a))

    返回input 张量中的元素个数

  4. torch.linspace

    ​
    torch.linspace(start, end, steps=100, out=None) → Tensor
    
    返回一个1维张量,包含在区间start 和 end 上均匀间隔的steps个点。 输出1维张量的长度为steps。
    
    参数:
    
    start (float) – 序列的起始点
    end (float) – 序列的最终值
    steps (int) – 在start 和 end间生成的样本数
    out (Tensor, optional) – 结果张量
    
    
    例子
    
    >>> torch.linspace(3, 10, steps=5)
    
      3.0000
      4.7500
      6.5000
      8.2500
     10.0000
    [torch.FloatTensor of size 5]
    
    ​
  5.  torch.ones

    torch.ones(*sizes, out=None) → Tensor
    返回一个全为1 的张量,形状由可变参数sizes定义。
    
    参数:
    
    sizes (int...) – 整数序列,定义了输出形状
    out (Tensor, optional) – 结果张量 例子:
    >>> torch.ones(2, 3)
    
     1  1  1
     1  1  1
    [torch.FloatTensor of size 2x3]
    
    >>> torch.ones(5)
    
     1
     1
     1
     1
     1
    [torch.FloatTensor of size 5]

  6. torch.rand

    torch.rand(*sizes, out=None) → Tensor
    返回一个张量,包含了从区间[0,1)的均匀分布中抽取的一组随机数,形状由可变参数sizes 定义。
    
    参数:
    
    sizes (int...) – 整数序列,定义了输出形状
    out (Tensor, optinal) - 结果张量 例子:
    >>> torch.rand(4)
    
     0.9193
     0.3347
     0.3232
     0.7715
    [torch.FloatTensor of size 4]
    
    >>> torch.rand(2, 3)
    
     0.5010  0.5140  0.0719
     0.1435  0.5636  0.0538
    [torch.FloatTensor of size 2x3]
  7. torch.randn

    torch.randn(*sizes, out=None) → Tensor
    返回一个张量,包含了从标准正态分布(均值为0,方差为 1,即高斯白噪声)中抽取一组随机数,形状由可变参数sizes定义。 参数:
    
    sizes (int...) – 整数序列,定义了输出形状
    out (Tensor, optinal) - 结果张量
    例子::
    
    >>> torch.randn(4)
    
    -0.1145
     0.0094
    -1.1717
     0.9846
    [torch.FloatTensor of size 4]
    
    >>> torch.randn(2, 3)
    
     1.4339  0.3351 -1.0999
     1.5458 -0.9643 -0.3558
    [torch.FloatTensor of size 2x3]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值