mxnet slice_axis

mx.symbol.slice_axis可以直接在某一维上切割，选择整行或整列。

pick 是精准挑选指定位置的值，

mxnet slice_axis

begin end参数， 左开右闭，左边包含，右边不包含

  

   

    

     

    
    

     

      import mxnet as mx
     
    
   

    

     

    
    

     

      a = mx.nd.arange(18).reshape((3,2,3))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(a.shape)
     
    
   

    

     

    
    

     

      print(a)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      d = mx.nd.slice_axis(a, axis=0, begin=0, end=2)
     
    
   

    

     

    
    

     

      data=d.asnumpy()
     
    
   

    

     

    
    

     

      print(data)
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/116932277