Searched defs:flatten_forward_func (Results 1 – 2 of 2) sorted by relevance
24 def flatten_forward_func(x): function
33 def flatten_forward_func(x, start_dim=0, end_dim=-1): function