Struct leaf::layers::utility::reshape::ReshapeConfig
[−]
[src]
pub struct ReshapeConfig { pub shape: Vec<usize>, }
Specifies configuration parameters for a Reshape Layer.
Fields
shape | The target shape that the input should assume. Preceding dimensions are treated as independent inputs Defaults to |
Methods
impl ReshapeConfig
fn of_shape(shape: &[usize]) -> ReshapeConfig
Create a ReshapeConfig that describes a Reshape layer with a provided shape.