이 페이지는 Cloud Translation API를 통해 번역되었습니다.
Switch to English

tf.keras.backend.gather

TensorFlow 1 개 버전 GitHub의에서 소스보기

인덱스의 요소 취득 indices 텐서에 reference .

reference 텐서.
indices 인덱스의 정수 텐서.

동일 유형의 텐서 reference .

예를 들면 :

var = tf.keras.backend.variable([[1, 2, 3], [4, 5, 6]])
tf.keras.backend.eval(var)
array([[1., 2., 3.],
       [4., 5., 6.]], dtype=float32)
var_gathered = tf.keras.backend.gather(var, [0])
tf.keras.backend.eval(var_gathered)
array([[1., 2., 3.]], dtype=float32)
var_gathered = tf.keras.backend.gather(var, [1])
tf.keras.backend.eval(var_gathered)
array([[4., 5., 6.]], dtype=float32)
var_gathered = tf.keras.backend.gather(var, [0,1,0])
tf.keras.backend.eval(var_gathered)
array([[1., 2., 3.],
       [4., 5., 6.],
       [1., 2., 3.]], dtype=float32)