pytorch中常用的函数
本文整理了笔者在学习pytorch中经常遇到的一些函数,本篇博客会不断进行更新,并且会加上自己使用背景和使用经验。
1. torch.max()函数
笔者最近在学习目标检测的相关知识,无论是在计算多个bounding box之间的IOU还是确定bounding box的类别信息的时候,都会用到torch.max()函数。torch.max()可以得到一个tensor某个维度的最大值,可以的得到两个tensor之间对应元素之间的最大值。
这个函数的签名是这样的:
torch.max(input, d
原创
2020-06-12 17:35:45 ·
1137 阅读 ·
2 评论