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

由 Carrie 创建, 最后一次修改 2019-03-11

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)
以上内容是否对您有帮助:

二维码
建议反馈
二维码