pytorch 实现tensor与numpy数组转换

 更新时间:2019年12月27日 15:52:30   作者:头发光了你就强了  
今天小编就为大家分享一篇使用pytorch 实现tensor与numpy数组转换,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

看代码,tensor转numpy:

a = torch.ones(2,2)
b = a.numpy()
c=np.array(a) #也可以转numpy数组
print(type(a))
print(type(b))
print(a)
print(b)

输出为:

<class ‘torch.Tensor'>
<class ‘numpy.ndarray'>
tensor([[1., 1.],
[1., 1.]])
[[1. 1.]
[1. 1.]]

numpy转tensor:

import torch
import numpy as np

a = np.ones(5)
b = torch.from_numpy(a)
c=torch.Tensor(a) #也可以转pytorch Tensor
print(type(a))
print(type(b))
print(a)
print(b)

输出为:

<class ‘numpy.ndarray'>
<class ‘torch.Tensor'>
[1. 1. 1. 1. 1.]
tensor([1., 1., 1., 1., 1.], dtype=torch.float64)

可见pytorch的tensor对象与numpy数组是可以相互转换的,且numpy数组的默认类型是double

以上这篇pytorch 实现tensor与numpy数组转换就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • python能做什么 python的含义

    python能做什么 python的含义

    在本篇文章里小编给大家整理的是关于python能做什么 python的含义的相关知识点,有需要的朋友们学习下。
    2019-10-10
  • selenium+python自动化78-autoit参数化与批量上传功能的实现

    selenium+python自动化78-autoit参数化与批量上传功能的实现

    这篇文章主要介绍了selenium+python自动化78-autoit参数化与批量上传,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-03-03
  •  python中字符串的常见操作总结(二)

     python中字符串的常见操作总结(二)

    这篇文章主要介绍了python中字符串的常见操作,文章围绕主题展开详细的内容介绍,具有一定的参考价值,需要的小伙伴可以参考一下
    2022-07-07
  • 基于python实现生成指定大小txt文档

    基于python实现生成指定大小txt文档

    这篇文章主要介绍了基于python实现生成指定大小txt文档,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-07-07
  • selenium在执行phantomjs的API并获取执行结果的方法

    selenium在执行phantomjs的API并获取执行结果的方法

    今天小编就为大家分享一篇selenium在执行phantomjs的API并获取执行结果的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12
  • Pandas统计计数value_counts()的使用

    Pandas统计计数value_counts()的使用

    本文主要介绍了Pandas统计计数value_counts()的使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-07-07
  • Python 多线程抓取图片效率对比

    Python 多线程抓取图片效率对比

    Python由于有全锁局的存在,并不能利用多核优势。所以,如果你的多线程进程是CPU密集型的,那多线程并不能带来效率上的提升,相反还可能会因为线程的频繁切换,导致效率下降;如果是IO密集型,多线程进程可以利用IO阻塞等待时的空闲时间执行其他线程,提升效率。
    2016-02-02
  • Python必备基础之闭包和装饰器知识总结

    Python必备基础之闭包和装饰器知识总结

    都2021年了Python的闭包和装饰器难道你还不会?今天就带大家详细总结一下Python闭包和装饰器的相关知识,需要的朋友可以参考下
    2021-06-06
  • 深度总结Python中字符串的使用

    深度总结Python中字符串的使用

    本文主要来学习字符串数据类型相关知识,包括讨论如何声明字符串数据类型,字符串数据类型与 ASCII 表的关系等内容,超级干货,不容错过
    2023-08-08
  • python解析库Beautiful Soup安装的详细步骤

    python解析库Beautiful Soup安装的详细步骤

    Beautiful Soup是python的一个库,最主要的功能是从网页抓取数据,下面这篇文章主要给大家介绍了关于python解析库Beautiful Soup安装的详细步骤,文中通过图文介绍的非常详细,需要的朋友可以参考下
    2023-04-04

最新评论