tf-keras-skeleton.py 文件源码

def convert_kernel(kernel):
          """Converts a Numpy kernel matrix from Theano format to TensorFlow format.

          Also works reciprocally, since the transformation is its own inverse.

          Arguments:
              kernel: Numpy array (3D, 4D or 5D).

          Returns:
              The converted kernel.

          Raises:
              ValueError: in case of invalid kernel shape or invalid data_format.
          """
          kernel = np.asarray(kernel)
          if not 3 <= kernel.ndim <= 5:
            raise ValueError('Invalid kernel shape:', kernel.shape)
          slices = [slice(None, None, -1) for _ in range(kernel.ndim)]
          no_flip = (slice(None, None), slice(None, None))
          slices[-2:] = no_flip
          return np.copy(kernel[slices])