brainpy.math.flatten
brainpy.math.flatten#
- brainpy.math.flatten(input, start_dim=None, end_dim=None)[source]#
Flattens input by reshaping it into a one-dimensional tensor. If
start_dim
orend_dim
are passed, only dimensions starting withstart_dim
and ending withend_dim
are flattened. The order of elements in input is unchanged.Note
Flattening a zero-dimensional tensor will return a one-dimensional view.