Skip to content

Latest commit

 

History

History
22 lines (14 loc) · 905 Bytes

extract_link_features.md

File metadata and controls

22 lines (14 loc) · 905 Bytes

dragnn_ops.extract_link_features(handle, channel_id, component=None, name=None)

dragnn_ops.extract_link_features(handle, channel_id, component=None, name=None)

Defined in tensorflow/dragnn/core/ops/gen_dragnn_ops.py.

Given a handle to a ComputeSession and a channel index, outputs link features.

Output indices have shape {batch_size * beam_size * channel_size}.

Args:

  • handle: A Tensor of type string. handle to a ComputeSession.
  • channel_id: An int. feature channel to extract features for.
  • component: An optional string. Defaults to "".
  • name: A name for the operation (optional).

Returns:

A tuple of Tensor objects (step_idx, idx). * step_idx: A Tensor of type int32. step indices to read activations from. * idx: A Tensor of type int32. indices within a step to read activations from.