torch.from_numpy()函数(pytorch版)

torch.from_numpy(ndarray) 的作用就是将生成的数组(array)转换为张量Tensor。

该方法等同于torch.Tensor(ndarray)。

举例说明:

单纯的torch.from_numpy(ndarray)方法。

复制代码
import numpy
import torch

data1 = numpy.array([5, 6, 9])
print('data1的数据类型为:', type(data1))
print('data1的值为:', data1)

data2 = torch.from_numpy(data1)
print('data2的数据类型为:', type(data2))
print('data2的值为:', data2)

data2[1] = 3
print('data2的数据类型为:', type(data2))
print('data2的值为:', data2)

结果输出:

复制代码
data1的数据类型为: <class 'numpy.ndarray'>
data1的值为: [5 6 9]
data2的数据类型为: <class 'torch.Tensor'>
data2的值为: tensor([5, 6, 9], dtype=torch.int32)
data2的数据类型为: <class 'torch.Tensor'>
data2的值为: tensor([5, 3, 9], dtype=torch.int32)

使用torch.Tensor(ndarray)方法:

复制代码
import numpy
import torch

data1 = numpy.array([5, 6, 9])
print('data1的数据类型为:', type(data1))
print('data1的值为:', data1)

data3 = torch.Tensor(data1)
print('data3的数据类型为:', type(data3))
print('data3的值为:', data3)

输出结果:

复制代码
data1的数据类型为: <class 'numpy.ndarray'>
data1的值为: [5 6 9]
data2的数据类型为: <class 'torch.Tensor'>
data2的值为: tensor([5., 6., 9.])
相关推荐
Learn Beyond Limits6 分钟前
The learning process of Decision Tree Model|决策树模型学习过程
人工智能·深度学习·神经网络·学习·决策树·机器学习·ai
AI360labs_atyun8 分钟前
2025世界智博会,揭幕AI触手可及的科幻生活
人工智能·ai·音视频·生活
luoganttcc12 分钟前
小鹏汽车 vla 算法最新进展和模型结构细节
人工智能·算法·汽车
算家计算15 分钟前
面壁智能开源多模态大模型——MiniCPM-V 4.5本地部署教程:8B参数开启多模态“高刷”时代!
人工智能·开源
居然JuRan15 分钟前
从零开始学大模型之大语言模型
人工智能
扑克中的黑桃A17 分钟前
AI 对话高效输入指令攻略(一):了解AI对话指令
人工智能
算家计算29 分钟前
不止高刷!苹果发布会AI功能全面解析:实时翻译、健康监测重磅升级
人工智能·apple·资讯
m0_6770343542 分钟前
机器学习-异常检测
人工智能·深度学习·机器学习
张子夜 iiii1 小时前
实战项目-----在图片 hua.png 中,用红色画出花的外部轮廓,用绿色画出其简化轮廓(ε=周长×0.005),并在同一窗口显示
人工智能·pytorch·python·opencv·计算机视觉
胡耀超1 小时前
3.Python高级数据结构与文本处理
服务器·数据结构·人工智能·windows·python·大模型