W3cubDocs

/TensorFlow Python

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.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/contrib/kfac/utils/mat2d_to_layer_params