在如今互联网大厂面试环节中，现场手撕代码已经成为了每位面试者必须要经历的一关。目前85%以上的互联网公司都将面试者的现场编程能力作为能否拿到Offer的核心评价指标。如果可以在规定时间内流畅、顺利完成面试官给出题目的候选人，拿到高质量面评的概率要比其他人高出很多。

本文作为《一站式智驾感知算法求职宝典》专栏中CV&感知算法面试常见手撕代码题目汇总系列的第三篇文章，本篇内容包含：Softmax函数实现、Focal Loss损失函数实现、多矩形框水平IoU实现、Dropout层实现四道算法手撕代码题目，并给出了参考实现代码。

CV&感知算法面试常见手撕代码题目汇总系列将会持续为大家收录和整理互联网大厂面试中的高频手撕算法题目，并且给出参考实现代码，便于同学们实习和求职前复习，从而在面试手撕代码环节更加游刃有余！

《一站式智驾感知算法求职宝典》专栏包含诸多高质量求职内容系列，并且包含：

• 如何制作一份高质量个人简历保姆级教程
• 如何包装论文成果&科研项目&实习工作话术
• 十五家自动驾驶感知算法面经（附参考答案）
• 互联网大厂在线笔试ACM模式常见输入输出类型题目汇总（附Python和C++两种语言参考答案）
• CV&感知算法面试常见手撕代码题目汇总

无论你是即将开始秋招的应届毕业生，还是打算找实习的算法小白，《一站式智驾感知算法求职宝典》都将是你求职路上的得力助手！

一、Softmax函数实现

题目要求：实现Softmax函数

import torch
import torch.nn.functional as F

def custom_softmax(x, dim=None):
    x_exp = torch.exp(x)
    x_sum = x_exp.sum(dim=dim)
    
    return x_exp / x_sum

if __name__ == "__main__":
    torch.manual_seed(0)

    x = torch.randn(10)

    # custom softmax
    print(f"custom softmax output: \n{custom_softmax(x, dim=-1)}")

    # official softmax
    print(f"official softmax output: \n{F.softmax(x, dim=-1)}")
    

二、Focal Loss损失函数实现

题目要求：实现2D目标检测、BEV感知中常用的Focal Loss损失函数

import torch
import torch.nn as nn
import torch.nn.functional as F 

class FocalLoss(nn.Module):
    def __init__(self, alpha=0.25, gamma=2., reduction='mean'):
        super(FocalLoss, self).__init__()
        self.alpha = alpha
        self.gamma = gamma
        self.reduction = reduction

    def forward(self, pred_logit, gt_label):
        ce_loss = F.binary_cross_entr