PyTorch_标量张量和数字的转换

对于只有一个元素的张量,使用 item() 函数将该值从张量中提取出来。


代码

python 复制代码
import torch 
import numpy as np 

def test01():
    tensor1 = torch.tensor(30)
    tensor2 = torch.tensor([30])
    tensor3 = torch.tensor([[30]])

    print(tensor1.item())
    print(tensor2.item())
    print(tensor3.item())

    # 注意:张量中只有一个元素,如果有多个元素的话,使用 item 函数可能会报错
    # tensor4 = torch.tensor([30, 40])
    # print(tensor4.item())


if __name__ == "__main__":
    test01() 
相关推荐
这里有鱼汤11 小时前
小白必看:QMT里的miniQMT入门教程
后端·python
Baihai_IDP11 小时前
AI Agents 能自己开发工具自己使用吗?一项智能体自迭代能力研究
人工智能·面试·llm
大模型真好玩12 小时前
大模型工程面试经典(七)—如何评估大模型微调效果?
人工智能·面试·deepseek
黎燃20 小时前
短视频平台内容推荐算法优化:从协同过滤到多模态深度学习
人工智能
TF男孩21 小时前
ARQ:一款低成本的消息队列,实现每秒万级吞吐
后端·python·消息队列
飞哥数智坊21 小时前
多次尝试用 CodeBuddy 做小程序,最终我放弃了
人工智能·ai编程
后端小肥肠1 天前
别再眼馋 10w + 治愈漫画!Coze 工作流 3 分钟出成品,小白可学
人工智能·aigc·coze
唐某人丶1 天前
教你如何用 JS 实现 Agent 系统(2)—— 开发 ReAct 版本的“深度搜索”
前端·人工智能·aigc
FIT2CLOUD飞致云1 天前
九月月报丨MaxKB在不同规模医疗机构的应用进展汇报
人工智能·开源
阿里云大数据AI技术1 天前
【新模型速递】PAI-Model Gallery云上一键部署Qwen3-Next系列模型
人工智能