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.]]))
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())