机器学习 - 手动实现 ReLU 和 Sigmoid

最新推荐文章于 2024-05-02 20:19:05 发布

CHNMSCS

最新推荐文章于 2024-05-02 20:19:05 发布

阅读量272

点赞数 2

分类专栏：机器学习文章标签：机器学习深度学习人工智能 python

本文链接：https://blog.csdn.net/bschn123/article/details/137158086

版权

机器学习专栏收录该内容

25 篇文章 0 订阅

订阅专栏

直接上代码

import torch
import matplotlib.pyplot as plt

A = torch.arange(-10, 10, 1, dtype=torch.float(32))
def relu(x):
	return torch.maximum(torch.tensor(0), x)
plt.plot(relu(A))

结果如下：

import torch
import matplotlib.pyplot as plt

A = torch.arange(-10, 10, 1, dtype=torch.float(32))
def sigmoid(x):
	return 1 / (1 + torch.exp(-x))
	
plt.plot(sigmoid(A))