tf.shape()

tf.shape

tf.shape(
    input,
    name=None,
    out_type=tf.int32
)

例如:

  • 将矩阵的维度输出为一个维度矩阵
import tensorflow as tf
import numpy as np

A = np.array([[[1, 1, 1],
               [2, 2, 2]],
              [[3, 3, 3],
               [4, 4, 4]],
              [[5, 5, 5],
               [6, 6, 6]]])

t = tf.shape(A)
with tf.Session() as sess:
    print(sess.run(t))

# 输出
[3 2 3]

参数

  • input:张量或稀疏张量
  • name:op 的名字,用于tensorboard中
  • out_type:默认为tf.int32

返回值

  • 返回out_type类型的张量
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页