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() 
相关推荐
编程零零七10 分钟前
Python巩固训练——第一天练习题
开发语言·python·python基础·python学习·python练习题
吹风看太阳18 分钟前
机器学习16-总体架构
人工智能·机器学习
Zonda要好好学习31 分钟前
Python入门Day4
java·网络·python
moonsims1 小时前
全国产化行业自主无人机智能处理单元-AI飞控+通信一体化模块SkyCore-I
人工智能·无人机
MUTA️1 小时前
ELMo——Embeddings from Language Models原理速学
人工智能·语言模型·自然语言处理
海豚调度1 小时前
Linux 基金会报告解读:开源 AI 重塑经济格局,有人失业,有人涨薪!
大数据·人工智能·ai·开源
小龙在山东1 小时前
Python 包管理工具 uv
windows·python·uv
T__TIII1 小时前
Dify 插件非正式打包
人工智能
jerwey1 小时前
大语言模型(LLM)按架构分类
人工智能·语言模型·分类
IT古董1 小时前
【第三章:神经网络原理详解与Pytorch入门】02.深度学习框架PyTorch入门-(5)PyTorch 实战——使用 RNN 进行人名分类
pytorch·深度学习·神经网络