zoukankan      html  css  js  c++  java
  • 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()函数

  • 相关阅读:
    MathML
    Redux counterpart rematch dva
    flow
    pauseable 库
    a simple machine learning system demo, for ML study.
    react图工具集成
    SQLite
    Celery
    RabbitMQ installation
    Thunk
  • 原文地址:https://www.cnblogs.com/kelvin-liu/p/14292850.html
Copyright © 2011-2022 走看看