tf.contrib.kfac.utils.mat2d_to_layer_params

tf.contrib.kfac.utils.mat2d_to_layer_params(
    vector_template,
    mat2d
)

Defined in tensorflow/contrib/kfac/python/ops/utils.py.

Converts a canonical 2D matrix representation back to a vector.

Args:

  • vector_template: A Tensor or pair of Tensors shaped like layer parameters.
  • mat2d: A 2D Tensor with the same shape as the value of layer_params_to_mat2d(vector_template).

Returns:

A Tensor or pair of Tensors with the same coefficients as mat2d and the same shape as vector_template.