脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → pytorch tensor维度信息

pytorch 获取tensor维度信息示例

 更新时间：2020年01月03日 15:47:25   作者：siqi_fighting  
今天小编就为大家分享一篇pytorch 获取tensor维度信息示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

我就废话不多说了，直接上代码吧！

>>> import torch
>>> from torch.autograd import Variable
>>> from torch import IntTensor
>>> var = Variable(IntTensor([[1,0],[0,1]]))
 
>>> var
Variable containing:
 1 0
 0 1
[torch.IntTensor of size 2x2]
 
>>> var.size()
torch.Size([2, 2])
 
>>> list(var.size())
[2, 2]

参考https://stackoverflow.com/questions/46826218/pytorch-how-to-get-the-shape-of-a-tensor-as-a-list-of-int

以上这篇pytorch 获取tensor维度信息示例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • python去除字符串中换行符的四种方法

    python去除字符串中换行符的四种方法

    python中换行符的作用就是将字符串划分为上下两行去输出显示,但是在有些情况下是不需要去用到换行符的,这个时候就要用方法将其给去除掉,这一篇文章小编会来给大家介绍一下python去除字符串中换行符的方法,感兴趣的话就一起往下看看吧
    2024-04-04
  • Python3自定义json逐层解析器代码

    Python3自定义json逐层解析器代码

    这篇文章主要介绍了Python3自定义json逐层解析器代码，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-05-05
  • 回归预测分析python数据化运营线性回归总结

    回归预测分析python数据化运营线性回归总结

    本文主要介绍了python数据化运营中的线性回归一般应用场景，常用方法，回归实现，回归评估指标，效果可视化等，并采用了回归预测分析的数据预测方法
    2021-08-08
  • Python 获取当前所在目录的方法详解

    Python 获取当前所在目录的方法详解

    本文给大家讲解的是使用python获取当前所在目录的方法以及相关示例，非常的清晰简单，有需要的小伙伴可以参考下
    2017-08-08
  • 利用python进行数据加载

    利用python进行数据加载

    今天给大家带来的是关于Python的相关知识,文章围绕着python数据加载展开,文中有非常详细的介绍及代码示例,需要的朋友可以参考下
    2021-06-06
  • Keras存在自定义loss或layer怎样解决load_model报错问题

    Keras存在自定义loss或layer怎样解决load_model报错问题

    这篇文章主要介绍了Keras存在自定义loss或layer怎样解决load_model报错问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-09-09
  • Python使用DEAP库实现简单遗传算法

    Python使用DEAP库实现简单遗传算法

    DEAP是一个用于快速原型设计和实验的进化计算框架,它支持多种进化算法,包括遗传算法、遗传编程等,下面我们来看看如何使用DEAP库实现简单遗传算法吧
    2025-01-01
  • Python中一些深不见底的“坑”

    Python中一些深不见底的“坑”

    这篇文章主要给大家介绍了关于Python中一些深不见底的“坑”，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
    2019-06-06
  • Python实现人脸识别的详细图文教程

    Python实现人脸识别的详细图文教程

    人脸识别是人工智能的一个重要应用,下面这篇文章主要给大家介绍了关于Python实现人脸识别的相关资料,文中通过图文介绍的非常详细,需要的朋友可以参考下
    2022-08-08
  • Tensorflow 多线程设置方式

    Tensorflow 多线程设置方式

    今天小编就为大家分享一篇Tensorflow 多线程设置方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-02-02

最新评论