Python tf.eye用法及代码示例

构造一个单位矩阵或一批矩阵。

用法

tf.eye(
    num_rows, num_columns=None, batch_shape=None, dtype=tf.dtypes.float32, name=None
)

参数

num_rows 非负 int32 标量 Tensor 给出每个批处理矩阵中的行数。
num_columns 可选非负 int32 标量 Tensor 给出每个批处理矩阵中的列数。默认为 num_rows 。
batch_shape Python 整数的列表或元组或一维 int32 Tensor 。如果提供，返回的 Tensor 将具有此形状的前导批次尺寸。
dtype 结果Tensor中元素的类型
name 此 Op 的名称。默认为"eye"。

形状batch_shape + [num_rows, num_columns]的Tensor

另见tf.ones、tf.zeros、tf.fill、tf.one_hot。

# Construct one identity matrix.
tf.eye(2)
==> [[1., 0.],
     [0., 1.]]

# Construct a batch of 3 identity matrices, each 2 x 2.
# batch_identity[i,:,:] is a 2 x 2 identity matrix, i = 0, 1, 2.
batch_identity = tf.eye(2, batch_shape=[3])

# Construct one 2 x 3 "identity" matrix
tf.eye(2, num_columns=3)
==> [[ 1.,  0.,  0.],
     [ 0.,  1.,  0.]]

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.eye。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法

参数

返回