niftynet.layer.reshape module

class ReshapeLayer(output_size, name='reshaper')[source]

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)[source]