tf.rank函数:返回TensorFlow张量的秩

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

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

参见指南:张量转换>形状的确定和改变

返回张量的秩.

返回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

注意:张量的秩与矩阵的秩不一样.张量的秩是唯一选择张量的每个元素所需的索引的数量.秩也被称为 “order”,“degree” 或 “ndims”.

参数:

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

返回:

int32 类型的 Tensor.

numpy兼容性

相当于np.ndim

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号