Esta página foi traduzida pela API Cloud Translation.
Switch to English

tf.eye

TensorFlow 1 versão Ver fonte no GitHub

Construir uma matriz de identidade, ou um lote de matrizes.

Usado nos cadernos

Usado na guia Usado nos tutoriais

Veja também 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.]]
 

num_rows Não-negativo int32 escalar Tensor dando o número de filas em cada matriz lote.
num_columns Opcional não-negativo int32 escalar Tensor dando o número de colunas em cada matriz lote. O padrão é num_rows .
batch_shape Uma lista de números inteiros ou tuplo Python ou um 1-D int32 Tensor . Se fornecidos, o devolvido Tensor terá dimensões de lotes principais desta forma.
dtype O tipo de um elemento na resultante Tensor
name Um nome para este Op . O padrão é "olho".

Um Tensor de forma batch_shape + [num_rows, num_columns]