1. ホーム
  2. tensorflow

tensorflow 学習メモ (IX): tf.shape()とtensor.get_shape()

2022-02-16 09:46:40
<パス

どちらの方法も、変数の形状を取得することができます

tf.shape(x)

ここで、x はテンソルであってもなくてもよく、テンソルを返す。

shape=tf.placeholder(tf.float32, shape=[None, 227,227,3] )


こんな感じで来ることが多いです。 feed というデータがあり、実行時に知りたい場合は None を実行時に確認するためには tf.shape(x)[0] というのがあります。

tensor.get_shape()

tensorだけがこのメソッドを持っていて、中の値をpythonの型にしたタプルを返します。

tensor.shape

に相当します。 tensor.get_shape()