tf.TensorShape

tf.TensorShape은 텐서의 형태를 나타냅니다.

예제1

import tensorflow as tf

a = tf.constant([1, 2, 3])
b = tf.constant([[1, 2], [3, 4]])

print(a.shape)
print(type(a.shape))

print(b.shape)
print(type(b.shape))

(3,)
<class 'tensorflow.python.framework.tensor_shape.TensorShape'>
(2, 2)
<class 'tensorflow.python.framework.tensor_shape.TensorShape'>

shape은 텐서의 형태를 반환합니다.

TensorShape 객체임을 알 수 있습니다.

예제2

import tensorflow as tf

a = tf.constant([1, 2, 3])
b = tf.constant([[1, 2], [3, 4]])

print(a.shape.as_list())
print(b.shape.as_list())

[3]
[2, 2]

tf.TensorShape의 as_list() 메서드는 텐서의 형태를 리스트로 반환합니다.

이전글/다음글

이전글 : tf.rank

다음글 : tf.zeros_initializer

공유하기