import torch
X = torch.tensor([[1,0,0],[1,1,0],[1,0,1,],[1,1,1]] ,dtype = torch.float32)
nandgate = torch.tensor([[1],[1],[1],[0]],dtype = torch.float32)
w = torch.tensor([0.7,-0.5,-0.5] ,dtype = torch.float32) # b,w1,w2
def nanAdd(X,w):
zhat = torch.mv(X,w)
nandhat = torch.tensor([int(x) for x in zhat>=0],dtype = torch.float32) #int(True)=1,int(False)=0
return zhat,nandhat
zhat,nandhat = nanAdd(X,w)
print(zhat)
print(nandhat)
import matplotlib.pyplot as plt
import seaborn as sns
plt.style.use('seaborn-whitegrid') #设置图像的风格
sns.set_style("white")
plt.figure(figsize=(5,3)) #设置画布大小
plt.title("NAND AND GATE",fontsize=16) #设置图像标题
plt.scatter(X[:,1],X[:,2],c=nandgate,cmap="rainbow") #绘制散点图
plt.xlim(-1,3) #设置横纵坐标尺寸
plt.ylim(-1,3)
plt.grid(alpha=.4,axis="y") #显示背景中的网格
plt.gca().spines["top"].set_alpha(.0) #让上方和右侧的坐标轴被隐藏
plt.gca().spines["right"].set_alpha(.0);
import numpy as np
x = np.arange(-1,3,0.5)
plt.plot(x,(0.7-0.5*x)/0.5 ,color="k",linestyle="--");
与非门
最新推荐文章于 2023-08-04 10:07:17 发布