當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。