可视化学习笔记4-pytorch可视化激活函数（relu、relu6、leakyrelu、hardswish、Mish）代码

对视觉类论文详解（免费）感兴趣的同学，可以关注微信公众号 李卓璐随手记，将会不定期发布，注意查收哦~

源代码

import torch
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F


class ReLU(nn.Module):
    def __init__(self):
        super(ReLU, self).__init__()

    def forward(self, input)