Reshape

class torchelie.nn.Reshape(*shape)

Reshape the input volume

Parameters

  • *shape (ints) – new shape, WITHOUT specifying batch size as first

  • dimension –

  • it will remain unchanged. (as) –

forward(x)
training: bool