tf.rank函數(shù):返回TensorFlow張量的秩

2018-12-01 10:59 更新
tf.rank 函數(shù)
rank(
    input,
    name=None
)

定義在:tensorflow/python/ops/array_ops.py.

參見指南:張量轉(zhuǎn)換>形狀的確定和改變

返回張量的秩.

返回0維的 int32 Tensor,它表示 input 的秩.

例如:

# shape of tensor 't' is [2, 2, 3]
t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
tf.rank(t)  # 3

注意:張量的秩與矩陣的秩不一樣.張量的秩是唯一選擇張量的每個(gè)元素所需的索引的數(shù)量.秩也被稱為 “order”,“degree” 或 “ndims”.

參數(shù):

  • input:A Tensor或SparseTensor.
  • name:操作的名稱(可選).

返回:

int32 類型的 Tensor.

numpy兼容性

相當(dāng)于np.ndim

以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)