Flatten

class Flatten(**kwargs)

Flattens the input to two dimensional.

Inputs:

• data: input tensor with arbitrary shape (N, x1, x2, …, xn)

Output:

• out: 2D tensor with shape: (N, x1 cdot x2 cdot … cdot xn)