pytorch tensor转int_numpy和pytorch

pytorch tensor转int_numpy和pytorchtensor转换为numpy采用.numpy()函数即可a=torch.tensor([1,2,3])print(a)print(type(a))print(a.dtype)b=a.numpy()print(b)print(type(b))print(b.dtype)输出:torch.tensor整数默认为int64即LongTensor小数默认为float32不过一般对tensor采用tensor.data()或者tensor.detach(

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE稳定放心使用

tensor 转换为 numpy

采用 .numpy() 函数即可

a=torch.tensor([1,2,3])  
print(a)
print(type(a))
print(a.dtype)
b=a.numpy()
print(b)
print(type(b))
print(b.dtype)

输出:

在这里插入图片描述

torch.tensor 整数默认为 int64 即 LongTensor 小数默认为 float32
不过 一般对tensor 采用 tensor.data() 或者 tensor.detach() 来将变量脱离计算图,不计算梯度。

numpy 转换为 tensor

有两种函数
一种是
torch.from_numpy()

第二种是
torch.tensor() 其中用这种,还可以转换数据类型

a=np.array([1,2,3])
print(a)
print(type(a))
print(a.dtype)
b=torch.from_numpy(a)
# b=torch.tensor(a)
print(b)
print(type(b))
print(b.dtype)

在这里插入图片描述

Pytorch中定义了8种CPU张量类型和对应的GPU张量类型
torch.FloatTensor(2,3) #构建一个2*3 Float类型的张量 torch.float32
torch.DoubleTensor(2,3) #构建一个2*3 Double类型的张量 torch.float64
torch.HalfTensor (2,3) #构建一个2*3 HalfTenso类型的张量
torch.ByteTensor(2,3) #构建一个2*3 Byte类型的张量
torch.CharTensor(2,3) #构建一个2*3 Char类型的张量
torch.ShortTensor(2,3) #构建一个2*3 Short类型的张量
torch.IntTensor(2,3) #构建一个2*3 Int类型的张量 torch.int32
torch.LongTensor(2,3) #构建一个2*3 Long类型的张量 torch.int64

在这里插入图片描述

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/184269.html原文链接:https://javaforall.net

(0)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

  • ForkJoin 线程池[通俗易懂]

    ForkJoin 线程池[通俗易懂]一、分而治之严格来讲,分而治之不算一种模式,而是一种思想。它可以将一个大任务拆解为若干个小任务并行执行,提高系统吞吐量。主要讲两个场景,Master-Worker模式,ForkJoin线程池。ForkJoin线程池是jdk7之后引入的一个并行执行任务的框架,其核心思想也是将任务分割为子任务,有可能子任务还是很大,还需要进一步拆解,最终得到足够小的任务。将分割出来的子任务放入双端队列中,然后几个启动线程从双端队列中获取任务执行。子任务执行的结果放到一个队列里,另起线程从队列中获取数据,合并结果。

    2026年1月29日
    3
  • Rest和RPC接口区别「建议收藏」

    Rest和RPC接口区别「建议收藏」接口调用通常包含两个部分,序列化和通信协议。常见的序列化协议包括json、xml、hession、protobuf、thrift、text、bytes等;通信比较流行的是http、soap、websockect,RPC通常基于TCP实现,常用框架例如dubbo,netty、mina、thrift首先解释下两种接口调用:Rest:严格意义上说接口很规范,操作对象即为资源,对资源的四种操作(p…

    2022年8月31日
    2
  • 求delay函数的原理

    求delay函数的原理哪位大哥知道c语言delay函数的原理 

    2022年5月24日
    46
  • TinyXml库 使用方法

    TinyXml库 使用方法TinyXml下载链接:https://pan.baidu.com/s/1kXiTFSF使用TinyXML只需要将其中的6个文件拷贝到项目中就可以直接使用了,这六个文件是:tinyxml.h、tinystr.h、tinystr.cpp、tinyxml.cpp、tinyxmlerror.cpp、tinyxmlparser.cpp;TinyXml类介绍:    XmlBase:整个TinyXML模型的…

    2022年5月26日
    46
  • ubuntu、win跨平台局域网文件传输工具「建议收藏」

    ubuntu、win跨平台局域网文件传输工具「建议收藏」DuktoR6官网:https://www.msec.it/blog/dukto/win10不可使用NitroShare官网:https://nitroshare.net/win10可使用

    2022年6月4日
    36
  • java getmethods_method

    java getmethods_method小编典典如果您知道类型为List,则将其List.class用作参数。如果您事先不知道类型,请想象您有:publicvoidm(Listl){//alllists}publicvoidm(ArrayListl){//onlyarraylists}如果有任何自动方式,反射应调用哪种方法?如果需要,可以使用Class.getInterfaces()或,Class.getSup…

    2022年9月23日
    7

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号