阅读(5.1k) 书签 (0)

TensorFlow函数教程:tf.keras.backend.eye

2019-03-13 17:31 更新

tf.keras.backend.eye函数

tf.keras.backend.eye(
    size,
    dtype=None,
    name=None
)

定义在:tensorflow/python/keras/backend.py。

实例化一个单位矩阵并返回它。

参数:

  • size:整数,行数/列数。
  • dtype:字符串,返回的Keras变量的数据类型。
  • name:字符串,返回的Keras变量的名称。

返回:

返回一个Keras变量,一个单位矩阵。

示例: 

>>> from keras import backend as K 
>>> kvar = K.eye(3) 
>>> K.eval(kvar) array([[ 1., 0., 0.], 
                                       [ 0., 1., 0.], 
                                       [ 0., 0., 1.]], dtype=float32)