conv_general_dilated实现

参考：

https://github.com/DarrenZhang01/TensorFlow_GSoC/blob/2fd360c8b1b8c9106044034f6a8b5c2734db9c3d/tf_jax_stax/tf_conv_general.py

结果应该还不对：

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      This file contains a general conv operation for TensorFlow.
     
    
   

    

     

    
    

     

      Zhibo Zhang, 2020.06.06
     
    
   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      import tensorflow as tf
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # # Translate dimension numbers from number representations into string
     
    
   

    

     

    
    

     

      # # representations
     
    
   

    

     

    
    

     

      # #
     
    
   

    

     

    
    

     

      # # For example, in a 1D case, "NWC" means "# batches", "spatial dimensions" and
     
    
   

    

     

    
    

     

      # # "# Channels (features)", so the according number representations based on order
     
    
   

    

     

    
    

     

      # # "batch dimension, feature dimension, spatial dimension" is tuple (0, 2, 1);
     
    
   

    

     

    
    

     

      # #
     
    
   

    

     

    
    

     

      # # The reverse translation is simple. Set the default mapping
     
    
  
 

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

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