このページは Cloud Translation API によって翻訳されました。
Switch to English

tf.eye

TensorFlow 1つのバージョン GitHubの上のソースを表示

単位行列、または行列のバッチを作成します。

ノートPCで使用されます

ガイドで使用チュートリアルで使用されます

参照してくださいtf.onestf.zerostf.filltf.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 非負int32スカラーTensor各バッチの行列の行数を与えます。
num_columns 任意の非負int32スカラーTensor各バッチ行列の列の数を与えます。デフォルトはnum_rows
batch_shape Pythonの整数または1-Dのリストまたはタプルint32 Tensor 。提供されている場合、返されるTensor 、この形状のバッチ寸法をリードしています。
dtype 得られた要素のタイプTensor
name このため名前Op 。 「目」にデフォルト設定。

A Tensor形状のbatch_shape + [num_rows, num_columns]