pytorch crelu

维度0位batch_size,

  a = torch.cat((F.relu6(x), F.relu6(-x)), 1)

  

   

    

     

    
    

     

        m = nn.ReLU()
     
    
   

    

     

    
    

     

        input = torch.randn(2).unsqueeze(0)
     
    
   

    

     

    
    

     

        output = torch.cat((m(input),m(-input)))
     
    
  
 

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

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