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)