#NEWLINE# def infer_shape( self , node , shapes ) : #NEWLINE# #INDENT# ( x_shp , b_shp , idx_shp ) = shapes #NEWLINE# nll_shp = ( x_shp [ 0 ] , ) #NEWLINE# sm_shp = x_shp #NEWLINE# am_shp = idx_shp #NEWLINE# return [ self , sm_shp , am_shp ]