一起学PyTorch吧,PyTorch从入门开始 每天都更新 超详细 参数 常用方法 简单案列 共同学习,共同进步 坚持day1

本文详细介绍了PyTorch库中核心的张量操作API,包括创建、维度计算、数据类型设置,以及torchvision模块的基础使用。重点讲解了torch.tensor、torch.numel和torch.set_printoptions等关键函数,适合初学者和进阶者深入理解PyTorch的底层机制。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

主要分为两大部分:

主要包括PyTorch 常用的API:

        --torch:torch常用的API

        --torchvision:torchvision常用API

torch:

        torch包含了多维张量的数据结构以及基于其上的多种数据操作。他也提供了多种工具,其中一些可以更有效地张量和任意类型进行序列化。它有 CUDA 的对应实现,可以在 NVIDIA GPU 上进行张量运算 (计算能力 >=2.0)。

Tensors

        torch.is_tensor(obj):如果参数obj是一个PyTorch张量,则返回True。

        torch.is_storage(obj):如果obj是一个PyTorch storage对象,则返回Ture。

        torch.set_default_tensor_type(t):设置当前tensor的默认数据类型。

>>> torch.tensor([1.2, 3]).dtype # initial default for
floating point is torch.float32 , →
torch.float32
>>> torch.set_default_tensor_type(torch.DoubleTensor)
>>> torch.tensor([1.2, 3]).dtype # a new floating point
tensor , →
torch.float64

  torch.numel(input)->int:返回input张量中的元素个数。input(Tensor):输入张量。

>>> a = torch.randn(1,2,3,4,5) 
>>> torch.numel(a)
120
>>> a = torch.zeros(4,4) 
>>> torch.numel(a)
16
torch.set_printoptions(precision=None, threshold=None,
edgeitems=None, linewidth=None, profile=None)} 
wAAACH5BAEKAAAALAAAAAABAAEAAAICRAEAOw==
设置打印选项。完全参考自 Numpy
参数:
        -precision:浮点数输出的精度位数(默认为8)
        -threshold:阈值,触发汇总显示而不是完全显示(repr)的数组元素的总数(默认为1000)
        -edgeitems:汇总显示中,每维(轴)两端显示的项数(默认值为3)
        -linewidth:用于插入行间隔的每行字符数(默认为80)。Thresholded matricies will ignore this parameter。
        - profile pretty 打印的完全默认值。可以覆盖上述所有选项 ( 默认为 short, full)
        
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值