niftynet.layer.reshape module

class niftynet.layer.reshape.ReshapeLayer(output_size, name='reshaper')

Bases: niftynet.layer.base_layer.Layer

This class defines a simple reshape layer, principally for passing feature maps to fully connected layers.

layer_op(input_tensor)