3873029035
* added softmax activation layer * update docstring * reformatted * updates after review * updating missed method name update * update return statement * added logsumexp class and tests * update doc string
21 lines
539 B
Python
21 lines
539 B
Python
import tensorflow as tf
|
|
|
|
|
|
def segment_sum(data, segment_ids, num_segments=None, sorted=False):
|
|
if sorted:
|
|
return tf.math.segment_sum(data, segment_ids)
|
|
else:
|
|
return tf.math.unsorted_segment_sum(data, segment_ids, num_segments)
|
|
|
|
|
|
def top_k(x, k, sorted=True):
|
|
return tf.math.top_k(x, k, sorted=sorted)
|
|
|
|
|
|
def in_top_k(targets, predictions, k):
|
|
return tf.math.in_top_k(targets, predictions, k)
|
|
|
|
|
|
def logsumexp(x, axis=None, keepdims=False):
|
|
return tf.math.reduce_logsumexp(x, axis=axis, keepdims=keepdims)
|