Advertisement

《动手学深度学习》PyTorch实现Softmax与FashionMNIST数据集训练

阅读量:

以下内容将进行详细介绍:

  • 从基础开始构建softmax函数
  • 采用简明方式实现softmax算法

softmax函数从零实现

引入相关模块

复制代码
    # import needed package
    %matplotlib inline
    from IPython import display
    import matplotlib.pyplot as plt
    
    import torch
    import torchvision
    import torchvision.transforms as transforms
    import time
    
    import sys
    sys.path.append("/home/kesci/input")
    import d2lzh1981 as d2l
    
    print(torch.__version__)
    print(torchvision.__version__)
    
    
      
      
      
      
      
      
      
      
      
      
      
      
      
      
      
      

全部评论 (0)

还没有任何评论哟~