pytorch使用 to 进行类型转换方式
作者：月下花弄影 发布时间：2022-10-05 23:28:27
标签：pytorch,to,类型转换
在程序中，有多种方法进行强制类型转换。
本博文将介绍一个非常常用的方法：to（）方法。
我们通常使用它来进行GPU和CPU的类型转换，但其实也可以用来进行torch的dtype转换。
常见方法：tensor.to(‘cuda:0')
先看官网介绍：
**Performs Tensor dtype and/or device conversion. A torch.dtype and torch.device are inferred from the arguments of self.to(*args, kwargs).
本文举一个例子，将一个tensor转化成与另一个tensor相同的数据类型和相同GPU或CPU类型
import torch
device = 'cuda:0'
a = torch.zeros(2, 3)
print(type(a))
b = torch.ones(3, 4).to(device)
print(type(b))
c = torch.matmul(a, b)
print(type(c))
我们看到这个代码会出错的。因为a和b是不同的device，一个是CPU，一个是GPU，不能运行。
修改如下：
a = a.to(b)
d = torch.matmul(a, b)
print(type(d))
可以看到to还是很好用的，尤其是不确定我们的数据类型和device时。
其实pytorch中还有很多其他方法可以这么做，以后会继续介绍。
来源：https://blog.csdn.net/qq_27261889/article/details/100175612
0
投稿猜你喜欢
- 在 github 的 vue-router 中找到同样的一个问题：3.0.1版本通过router实例无法跳转昨天发现有些路由不能正常跳转，找
- 由于asp中是使用双引号作为字符串的开始和结束标志的，单一个字符串中的双引号出现次数大于两个时，程序就有可能运行错误。asp中是怎么输出引号
原始需求：例如有一个列表：l = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]希望把它转换成下面这种形式：[1, 2,- 简介Node2vec是一种用于图嵌入（Graph Embedding）的方法，可用于节点分类、社区发现和连接预测等任务。实现过程
- 关于CSS中的命名规则（其实我觉得是XHTML元素的命名规则更加合理些）这个问题，已经有很多人在说了，其中也不乏一些真知灼见。不过这种东西也
- 一主一从： Master: OS:centos release 5.6 DB:mysql 5.5.8 IP:192.168.1.2 Slav
- 今天淘到的，终于可以放心的使用IE7了，再也不用换回IE6做测试了！因为这个10M大小的软件已经包含了从IE3到IE6的版本，安装过程中可以
- PyCharm就是Python语言开发中一个很受欢迎的IDE，界面类似于visual studio，android studio，集成的功能
- tomorrow是我最近在用的一个爬虫利器，该模块属于第三方的一个模块，使用起来非常的方便，只需要用其中的threads方法作为装饰器去修饰
- 之前关于 Vue 数据绑定原理的一点分析，最近需要回顾，就顺便发到随笔上了在之前实现一个自己的Mvvm中，用 setter 来观测model
- CONVERT函数用于将值转换为指定的数据类型或字符集1.转换指定字符集CONVERT函数用于将字符串expr的字符集变成transcodi
- pip install指定安装目录pip install一般会有默认的安装目录，可以通过python -m site进行查看，当然也可以对默
- 本文实例讲述了Python minidom模块用法。分享给大家供大家参考，具体如下：一、DOM写XML文件# -*- coding:utf-
- 因为要用到过滤一组中重复的数据，使之变成没有重复的一组数据的功能，百度了一下，居然有朋友乱写，而且比较多，都没有认真测试过，只对字符可以，但
- 【需求背景】有时候我们要对比两份配置文件是不是一样，或者比较两个文本是否异样，可以使用linux命令行工具diff a_file b_fil
- 使用Ajax技术网页应用能够快速地将增量更新呈现在用户界面上，而不需要重载刷新整个页面，这使得程序能够更快地回应用户的操作，如下笔记将简单介
- 多线程-共享全局变量#coding=utf-8from threading import Threadimport timeg_num =
- 一直很想就搜索结果页写一些心得文章出来，甚至连目录都整理好了可是就是一直没有动手。因为总是觉得还差很多东西需要研究需要分析需要验证。最近也组
- 之前博客有用logstash-input-jdbc同步mysql数据到ElasticSearch，但是由于同步时间最少是一分钟一次，无法满足
- 当你的web应用程序成长到一定规模时，你可能需要使用消息队列来处理异步任务、事件或在多个服务之间传递消息。Kafka是一个开源的消息队列系统
