Python layer_params_to_mat2d Examples

Example #1

File: utils_test.py Project: leox1v/kfac

 def testFullyConnectedLayerParamsTensorToMat2d(self):
   with tf.Graph().as_default(), self.test_session() as sess:
     tf.set_random_seed(200)
     layer_params = self._fully_connected_layer_params()
     output = utils.layer_params_to_mat2d(layer_params[0])
     self.assertListEqual([2, 2], output.get_shape().as_list())
     self.assertAllClose(sess.run(output), np.array([[1., 2.], [4., 3.]]))

Example #2

File: utils_test.py Project: leox1v/kfac

 def testConvLayerParamsTupleToMat2d(self):
   with tf.Graph().as_default():
     tf.set_random_seed(200)
     layer_params = self._conv_layer_params()
     output = utils.layer_params_to_mat2d(layer_params)
     self.assertListEqual([2 * 2 * 3 + 1, 4], output.get_shape().as_list())