可视化学习笔记4-pytorch可视化激活函数(relu、relu6、leakyrelu、hardswish、Mish)代码
源代码importtorchimportmatplotlib.pyplotaspltimporttorch.nnasnnimporttorch.nn.functionalasFclassReLU(nn.Module):def__init__(self):super(ReLU,self).__init__()defforward(self,input):returnF.relu(input)#def