tensoflow 2.0 numpy. ndarray 와 tenor 가 직접 상호 전환 합 니 다.

298 단어 tensorflow
import numpy as np
import tensorflow as tf

a = np.random.random((5,3))
b = np.random.randint(0,9,(3,1))
c = tf.tensordot(a.astype(np.float),b.astype(np.float),axes=1)
# tensor  ndarray
dn = c.numpy()
print(dn)
# ndarray tensor
tn = tf.convert_to_tensor(dn)
print(tn)

좋은 웹페이지 즐겨찾기