Searched defs:flatten_backward_func (Results 1 – 2 of 2) sorted by relevance
29 def flatten_backward_func(x): function
42 def flatten_backward_func(x, start_dim=0, end_dim=-1): function