pytorch 统计tensor中总元素数量的函数:numel()

示例:

import torch
a = torch.randn(2,3,4)
print(a.numel())    # 24

统计model中所有可训练参数量:

num_params = sum(p.numel() for p in model.parameters())

注:numel() 是pytorch的函数,只适用于 tensor,不能用于统计 list、tuple、dict 等的元素数量。

  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
引用\[1\]:一维向量一般用来表示Bias或者是Linear input。\[1\] torch.tensor(\[1.1\]) \[2\] tensor(\[1.1000\]) \[3\] torch.tensor(\[1.1, 2.2\]) \[4\] tensor(\[1.1000, 2.2000\]) \[5\] torch.FloatTensor(1) # 创建一维Float型tensor \[6\] tensor(\[1.4013e-45\]) \[7\] torch.FloatTensor(2) \[8\] tensor(\[3.7140e+00, 4.5916e-41\]) \[9\] data = np.ones(2) # 通过np创建向量再转换成tensor \[10\] array(\[1., 1.\]) \[11\] data \[12\] torch.from_numpy(data) \[13\] tensor(\[1., 1.\], dtype=torch.float64) \[14\] a = torch.ones(2) \[15\] a.shape \[16\] torch.Size(\[2\]) 二维Tensor。引用\[2\]:0维tensor一般用来表示损失函数的数值及Loss的数值。\[1\] import torch \[2\] torch.tensor(1.) \[3\] tensor(1.) \[4\] a = torch.tensor(1.3) \[5\] a.shape \[6\] torch.Size(\[\]) \[7\] len(a.shape) \[8\] 0 \[9\] a.size() \[10\] torch.Size(\[\]) 一维向量。引用\[3\]:一般用来表示CNN的数据,表达形式为\[b,c,w,h\] b: batch size c: channel w: width h: hight \[1\] a = torch.rand(2,3,28,28) \[2\] a.shape \[3\] torch.Size(\[2, 3, 28, 28\]) \[4\] a.numel() \[5\] 4704 \[6\] a.dim() \[7\] 4 创建Tensor import from numpy \[1\] import torch \[2\] import numpy as np \[1\] a = np.array(\[2,3,3\]) \[4\] torch.from_numpy(a) \[5\] tensor(\[2, 3, 3\], dtype=torch.int32) \[6\] a = np.ones(\[2,3\]) # 创建元素都为1的2行3列矩阵。 \[7\] torch.from_numpy(a) \[8\] tensor(\[\[1., 1., 1.\], \[1., 1., 1.\]\], dtype=torch.float64) import from list \[1\] torch.tensor(\[2.,3.2\]) \[2\] tensor(\[2.0000, 3.2000\]) \[3\] torch.FloatTensor(\[2.,3.2\]) \[4\] tensor(\[2.0000, 3.2000\]) \[5\] torch.tensor(\[\[2.,3.2\],\[2.,2.3\]\]) \[6\] tensor(\[\[2.0000, 3.2000\], \[2.0000, 2.3000\]\]) # 创建两行三列的tensor。 \[7\] torch.FloatTensor(2,3) \[8\] tensor(\[\[0., 0., 0.\], \[0., 0., 0.\]\])。 问题: pytorch tensor 转string 回答: 在PyTorch,可以使用`str()`函数将一个PyTorch tensor转换为字符串。例如,如果有一个名为`tensor1`的tensor,可以使用`str(tensor1)`将其转换为字符串表示形式。 #### 引用[.reference_title] - *1* *2* *3* [Pytorch数据类型及Tensor的相关操作](https://blog.csdn.net/leeyns/article/details/113367690)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ctrl A_ctrl C_ctrl V

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值