好玩的人工智能
快乐的深度学习

告诉你的向量是几维,用tf.shape

Tensorflow官网给tf.shape的定义:

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

参数说明:
input: 张量或稀疏张量 A Tensor or SparseTensor.
name: 操作符名字,和Variable的name一样 A name for the operation (optional).
out_type:默认的输出类型是tf.Int32,也可以指定int64 (Optional) The specified output type of the operation (int32 or int64). Defaults to tf

写个demo试试:

#!/usr/bin/python
# -*- coding: utf8 -*-)

import tensorflow as tf

t = tf.constant([[[1, 1, 1], [2, 2, 2]], [[3, 3, 3], [4, 4, 4]]])
l = tf.constant([1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4]) 
s = tf.shape(t)

with tf.Session() as sess:
    print (sess.run(s))
    print (sess.run(l))

输出:

[2 2 3]
[1 1 1 2 2 2 3 3 3 4 4 4]

 

https://www.tensorflow.org/api_docs/python/tf/shape

未经允许不得转载:零点智能 » 告诉你的向量是几维,用tf.shape
分享到: 更多 (0)

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址

零点智能 人工智能社区,加Q群:469331966

投稿&建议&加Q群