Pytorch中的tensor操作

本文详细介绍了PyTorch中Tensor的各种操作,包括扩充维度、比较大小、torch.prod、重组操作、取整、张量拼接、切断反向传播途径、contiguous、图像维度转换、数据类型转换、tensor乘法、转置以及转换为list等,同时提供了相关函数的使用示例和解释。
摘要由CSDN通过智能技术生成

一. 扩充维度

1.1. 使用unsqueeze

1.2. YOLO里面使用的 [none]

wh1:

wh2:  

    wh1 = wh1[:, None]
    wh2 = wh2[None]

wh1:

wh2: 

二. 比较大小

2.1 torch.min(只需要最后一维shape相同即可)

3x1x2维的张量 与 1x3x2维的张量比较(使用torch.min(wh1,wh2))(下图右侧)

等效  为左侧扩充维度后的张量比较

已经验证了,结果相同。说明min的内部就是先做了扩充维度,再做比较。

2.2 torch.max

 _, predicted = torch.max(outputs.data, 1)

 返回结果本身以及索

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值