• Tensor的创建和维度的查看


    常见的Tensor创建方法

    1,基础Tensor函数:torch.Tensor(2,2)32位浮点型

    2,指定类型: torch.DoubleTensor(2,2)64位浮点型

    3,使用python的列表序列:torch.Tensor([[1,2],[3,4]])

    4,默认值为0 :torch.zeros(2,3)

    5,默认值为1 : torch.one(2,2)

    6,对角张量 : torch.eye(2,2)

    7,随机张量 :torch.randn(2,2)

    8,随机排列张量  :torch.randperm(4)长度为4的随机排列向量

    >>> a = torch.arange(1,5)
    >>> a
    tensor([1, 2, 3, 4])
    >>> a = torch.arange(1,5,2)
    >>> a
    tensor([1, 3])

    >>> a = torch.linspace(1,5,2)
    >>> a
    tensor([1., 5.])
    >>> a = torch.linspace(1,5,3)
    >>> a
    tensor([1., 3., 5.])
    #torch.tensor([1,2,3])

    tensor([1,2,3])

    维数的查看有两种  1,a.shape    2,a.size()

    查看tensor中的元素的个数 a.numel()和a.nelement()函数

  • 相关阅读:
    作业二
    作业一
    作业十一
    作业十
    作业九
    作业八
    作业七
    作业六
    作业五
    JAVA第三次作业
  • 原文地址:https://www.cnblogs.com/kelvin-liu/p/14292850.html
Copyright © 2020-2023  润新知