P2 Pytorch 张量数据类型
创始人
2024-02-29 08:13:33
0

前言

    1: 数据类型

    2: 常用API 

参考:

   课时14 张量数据类型-1_哔哩哔哩_bilibili


一   数据类型

        

         除了string ,相对于Numpy PyTorch 都能找到对应的数据类型

         

       1.1 常用的Data type

       

          常用的5种:

          IntTensor, LogTensor, ByteTensor, DoubleTensor, FloatTensor

        

# -*- coding: utf-8 -*-
"""
Created on Tue Nov 29 16:20:52 2022@author: chengxf2
"""
import torchdef checktype():a = torch.randn(2,3)print("\n\t type:",a.type())bFloat = isinstance(a, torch.FloatTensor)print("\n\t bFloat",bFloat)bDouble = isinstance(a, torch.cuda.DoubleTensor)print("\n\t bDouble: ",bDouble)print("\n\t sp",a.shape)dim = len(a.shape)print("\n\t 维度",dim)b = torch.tensor(1)print("\n\t b: ",b.type()) print("\n\t value ",b.item())print("\n\t sp",b.shape)if __name__ == "__main__":checktype()


二  常用API 函数

   

Function

说明

Dim()

 维度

Numel()

占用内存

shape

数组的每个维度的长度

Size(0)

对应维度的大小

FromNumpy()

Numpy to Pytorch

Numpy()

Pytorch to Numpy

item

取元素

torch.FloatTensor

Float 类型

torch.DoubleTensor

Double 类型

torch.IntTensor

Int 类型

torch.ByteTensor

Byte 类型

       

上一篇:Thread的基本用法

下一篇:flex布局

相关内容

热门资讯

监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
Fluent中创建监测点 1 概述某些仿真问题,需要创建监测点,用于获取空间定点的数据࿰...
educoder数据结构与算法...                                                   ...
MySQL下载和安装(Wind... 前言:刚换了一台电脑,里面所有东西都需要重新配置,习惯了所...
MFC文件操作  MFC提供了一个文件操作的基类CFile,这个类提供了一个没有缓存的二进制格式的磁盘...
有效的括号 一、题目 给定一个只包括 '(',')','{','}'...
【Ctfer训练计划】——(三... 作者名:Demo不是emo  主页面链接:主页传送门 创作初心ÿ...