#NEWLINE# def repeat_elements( x , rep , axis ) : #NEWLINE# #INDENT# 'Repeats the elements of a tensor along an axis, like np.repeat\n\n    If x has shape (s1, s2, s3) and axis=1, the output\n    will have shape (s1, s2 * rep, s3)\n    ' #NEWLINE# return T . repeat ( x , rep , axis = axis )