Skip to content

Instantly share code, notes, and snippets.

@3h4
Created November 19, 2016 11:37
Show Gist options
  • Save 3h4/3826b5db1a8fb7bf3e35b84904ba7505 to your computer and use it in GitHub Desktop.
Save 3h4/3826b5db1a8fb7bf3e35b84904ba7505 to your computer and use it in GitHub Desktop.
logits_series = tf.unpack(tf.reshape(logits, [batch_size, truncated_backprop_length, num_classes]), axis=1)
predictions_series = [tf.nn.softmax(logit) for logit in logits_list]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment