• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    pytorch 如何查看数据类型和大小

    问题描述:

    查看tensor数据大小时使用了data.shape(),报错:

    TypeError: 'torch.Size' object is not callable 或 TypeError: 'tuple' object is not callable。

    解决方法:

    查看数据类型:data.dtype

    查看数据大小:data.shape

    补充:pytorch tensor比较大小 数据类型要注意

    如下

    a = torch.tensor([[0, 0], [0, 0]])
    print(a>=0.5)

    输出

    tensor([[1, 1],
            [1, 1]], dtype=torch.uint8)

    结果明显不对, 分析原因是因为, a是long类型, 而0.5是float. 0.5会被转化为 long, 变为0. 因此结果会出错, 做出如下修改就可以得到正确答案

    正确用法:

    a = torch.tensor([[0, 0], [0, 0]]).float()
    print(a>=0.5)

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

    您可能感兴趣的文章:
    • PyTorch中Tensor的数据类型和运算的使用
    • pytorch逐元素比较tensor大小实例
    • pytorch常见的Tensor类型详解
    • pytorch中tensor张量数据类型的转化方式
    上一篇:Python数据分析之pandas比较操作
    下一篇:Pytorch 如何查看、释放已关闭程序占用的GPU资源
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    pytorch 如何查看数据类型和大小 pytorch,如何,查看,数据,类型,