■ Tensor 클래스의 get_shape 메소드를 사용해 랭크를 구하는 방법을 보여준다.
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import tensorflow as tf scalarTensor = tf.constant(100) vectorTensor = tf.constant([1, 2, 3, 4, 5]) matrixTensor = tf.constant([[1, 2, 3], [4, 5, 6]]) cubeMatrixTensor = tf.constant([[[1], [2], [3]], [[4], [5], [6]], [[7], [8], [9]]]) scalarTensorShape = scalarTensor.get_shape() vectorTensorShape = vectorTensor.get_shape() matrixTensorShape = matrixTensor.get_shape() cubeMatrixTensorShape = cubeMatrixTensor.get_shape() print(scalarTensorShape ) print(vectorTensorShape ) print(matrixTensorShape ) print(cubeMatrixTensorShape) [결과] () (5,) (2, 3) (3, 3, 1) |