[PYTHON/TENSORFLOW] convert_to_tensor 함수 : 텐서 구하기

icodebroker TENSORFLOW 2024-06-18

■ convert_to_tensor 함수를 사용해 텐서를 구하는 방법을 보여준다.

▶ 예제 코드 (PY)


import numpy as np
import tensorflow as tf

ndArray = np.array([[[ 0,  1,  2], [ 3,  4,  5], [ 6,  7,  8]],
                   [[ 8, 10, 11], [12, 13, 14], [15, 16, 17]],
                   [[18, 19, 20], [21, 22, 23], [24, 25, 26]]])

tensor = tf.convert_to_tensor(ndArray, dtype = tf.float64)

with tf.Session() as session:
    print(tensor.get_shape())
    print(session.run(tensor))

[결과]

(3, 3, 3)
[[[ 0.  1.  2.]
  [ 3.  4.  5.]
  [ 6.  7.  8.]]

 [[ 8. 10. 11.]
  [12. 13. 14.]
  [15. 16. 17.]]

 [[18. 19. 20.]
  [21. 22. 23.]
  [24. 25. 26.]]]

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

26

27

28

29

import numpy as np

import tensorflow as tf

ndArray = np.array([[[ 0, 1, 2], [ 3, 4, 5], [ 6, 7, 8]],

[[ 8, 10, 11], [12, 13, 14], [15, 16, 17]],

[[18, 19, 20], [21, 22, 23], [24, 25, 26]]])

tensor = tf.convert_to_tensor(ndArray, dtype = tf.float64)

with tf.Session() as session:

print(tensor.get_shape())

print(session.run(tensor))

[결과]

(3, 3, 3)

[[[ 0. 1. 2.]

[ 3. 4. 5.]

[ 6. 7. 8.]]

[[ 8. 10. 11.]

[12. 13. 14.]

[15. 16. 17.]]

[[18. 19. 20.]

[21. 22. 23.]

[24. 25. 26.]]]

AI PYTHON TENSORFLOW