当前位置:   article > 正文

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

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

直接上代码

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))

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

结果如下:
结果如下

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))

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9

结果如下

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/351200
推荐阅读
相关标签
  

闽ICP备14008679号