Python tensor_array_gather_v3 Exemples

Exemple #1

Fichier : tensor_array_ops.py Projet : aritratony/tensorflow

 def gather(self, indices, name=None):
   """See TensorArray."""
   if self._element_shape:
     element_shape = self._element_shape[0]
   else:
     element_shape = tensor_shape.unknown_shape(None)
   value = gen_data_flow_ops.tensor_array_gather_v3(
       handle=self._handle,
       indices=indices,
       flow_in=self._flow,
       dtype=self._dtype,
       name=name,
       element_shape=element_shape)
   if self._element_shape and self._element_shape[0].dims is not None:
     value.set_shape([None] + self._element_shape[0].dims)
   return value

Exemple #2

Fichier : tensor_array_ops.py Projet : sgcm520/tensorflow2

 def gather(self, indices, name=None):
     """See TensorArray."""
     if self._element_shape:
         element_shape = self._element_shape[0]
     else:
         element_shape = tensor_shape.TensorShape(None)
     value = gen_data_flow_ops.tensor_array_gather_v3(
         handle=self._handle,
         indices=indices,
         flow_in=self._flow,
         dtype=self._dtype,
         name=name,
         element_shape=element_shape)
     if self._element_shape and self._element_shape[0].dims is not None:
         value.set_shape([None] + self._element_shape[0].dims)
     return value