pytorch numpy,pytorch nn.softmax

  pytorch numpy,pytorch nn.softmax

  【pytorch】使用数组实现框架的分类器函数与交叉熵函数_安安爸克莉丝的技术博客_博客

  torch.nn.functional.F.softmax公式

  将数组作为铭牌导入

  进口火炬

  导入火炬. nn .功能为F

  def my_softmax(x):

  exp_x=np.exp(x)

  返回指数x/np.sum(指数x)

  x=np.array([1.5,2.2,3.1,0.9,1.2,1.7])

  xt=torch.from_numpy(x)

  print(F.softmax:,F.softmax(xt))

  print(my_softmax ,my_softmax(x))结果一致

  从源码上来看,torch.nn.functional.F.softmax实际上调用的是张量自身的分类器函数

  torch.nn.functional.F.softmax公式

  将数组作为铭牌导入

  进口火炬

  导入火炬. nn .功能为F

  def my_softmax(x):

  exp_x=np.exp(x)

  返回指数x/np.sum(指数x)

  def my_log_softmax(x):

  返回np.log(my_softmax(x))

  x=np.array([1.5,2.2,3.1,0.9,1.2,1.7])

  xt=torch.from_numpy(x)

  print(F.softmax:,F.log_softmax(xt))

  print(my_softmax ,my_log_softmax(x))结果一致

郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。

留言与评论(共有 条评论)
   
验证码: