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

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)
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部