PyTorch 张量数据类型

  • 【数据类型】Python 与 PyTorch 常见数据类型对应:

    a.type() 获取数据类型,用 isinstance(a, 目标类型) 进行类型合法化检测

    python3 复制代码
    >>> import torch
    >>> a = torch.randn(2,3)
    >>> a
    tensor([[-1.7818, -0.2472, -2.0684],
            [ 0.0117,  1.4698, -0.9359]])
    >>> a.type()  ## 获取数据类型
    'torch.FloatTensor'
    >>> isinstance(a, torch.FloatTensor)  ## 类型合法化检测
    True
    >>> 
  • 【什么是张量】标量与张量:用 a.dim(), a.shape 或者 a.size() 查看 dim 为 0 是标量,否则是张量

    python3 复制代码
    >>> import torch
    >>>
    >>> a = torch.tensor(1) 
    >>> a
    tensor(1)
    >>> a.dim()
    >>> 0  ## 标量
    
    >>> a = torch.Tensor([1]) 
    >>> a
    tensor([1.])
    >>> a.dim()
    >>> 1  ## 张量
  • 【生成张量】常见方法如下:

    • 常见随机方法:torch.randn(shape), torch.rand(shape), torch.randint(min, max, shape), torch.rand_like(a), torch.normal(mean, std) ... 具体示例如下

    • Dim 1 / rank 1: 以 size 2 为例

      python3 复制代码
      >>> a = torch.randn(2)   ## 随机,
      >>> a: tensor([1.4785, 0.6089])
      
      >>> a = torch.Tensor(2)   ## 接收维度, unintialized 不推荐
      >>> a: tensor([5.4086e+26, 4.5907e-41])
      >>> a = torch.Tensor([1,2])   ## 同 torch.tensor([1,2]) 接收具体数据
      >>> a: tensor([1, 2])
      
      >>> a = torch.from_numpy(np_data)  ## 数据维持不变,类型一一对应
      
      >>> a = torch.full([2],7)   ## 全部填充为一样的值   
      >>> a: tensor([7, 7])
      
      >>> a = torch.arange(0,10)   ## arange
      >>> a: tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
      
      >>> a = torch.linspace(0,10, steps=4)   ##  
      >>> a: tensor([ 0.0000,  3.3333,  6.6667, 10.0000])
      >>> a = torch.logspace(0,10, steps=4) 
      >>> a: tensor([1.0000e+00, 2.1544e+03, 4.6416e+06, 1.0000e+10])
    • Dim 2 / rank 2: 以 size [2,3] 为例

      python3 复制代码
      >>> a = torch.randn(2, 3)   ## 随机 
      >>> a: tensor([[ 2.0631, -1.7011,  0.6375],
      	    	   [-1.2104, -1.3341, -0.8187]])
      
      >>> a = torch.Tensor(2, 3)   ## 接收维度, unintialized 不推荐   
      >>> a: tensor([[-0.2438, -0.9554, -0.4694],
      			   [ 0.8636,  1.6497, -0.8862]])
      >>> a = torch.Tensor([[1,2,3],[4,5,6]])  ## 同 torch.tensor([[1,2,3],[4,5,6]]) 接收具体数据 
      >>> a: tensor([[1., 2., 3.],
             		   [4., 5., 6.]])
      
      >>> a = torch.from_numpy(np_data)  ## 数据维持不变,类型一一对应
      
      >>> a = torch.full([2,3],7)  ## 全部填充为一样的值
      >>> a: tensor([[7, 7, 7],
                     [7, 7, 7]])

      ...


相关推荐
智能汽车人3 分钟前
Robot---能打羽毛球的机器人
人工智能·机器人·强化学习
埃菲尔铁塔_CV算法4 分钟前
基于 TOF 图像高频信息恢复 RGB 图像的原理、应用与实现
人工智能·深度学习·数码相机·算法·目标检测·计算机视觉
ζั͡山 ั͡有扶苏 ั͡✾8 分钟前
AI辅助编程工具对比分析:Cursor、Copilot及其他主流选择
人工智能·copilot·cursor
东临碣石8211 分钟前
【AI论文】数学推理能否提升大型语言模型(LLM)的通用能力?——探究大型语言模型推理能力的可迁移性
人工智能·语言模型·自然语言处理
天天爱吃肉821817 分钟前
ZigBee通信技术全解析:从协议栈到底层实现,全方位解读物联网核心无线技术
python·嵌入式硬件·物联网·servlet
未来智慧谷40 分钟前
微软医疗AI诊断系统发布 多智能体协作实现疑难病例分析
人工智能·microsoft·医疗ai
野生技术架构师42 分钟前
简述MCP的原理-AI时代的USB接口
人工智能·microsoft
Allen_LVyingbo1 小时前
Python常用医疗AI库以及案例解析(2025年版、上)
开发语言·人工智能·python·学习·健康医疗
jndingxin1 小时前
OpenCV中超分辨率(Super Resolution)模块类cv::dnn_superres::DnnSuperResImpl
人工智能·opencv·dnn
NAGNIP1 小时前
一文搞懂FlashAttention怎么提升速度的?
人工智能·算法