| 98 | } |
| 99 | |
| 100 | func (l *LNormalization) GetKerasLayerConfig() interface{} { |
| 101 | inboundNodes := [][][]interface{}{ |
| 102 | {}, |
| 103 | } |
| 104 | for _, input := range l.inputs { |
| 105 | inboundNodes[0] = append(inboundNodes[0], []interface{}{ |
| 106 | input.GetName(), |
| 107 | 0, |
| 108 | 0, |
| 109 | map[string]bool{}, |
| 110 | }) |
| 111 | } |
| 112 | return jsonConfigLNormalization{ |
| 113 | ClassName: "Normalization", |
| 114 | Name: l.name, |
| 115 | Config: map[string]interface{}{ |
| 116 | "axis": l.axis, |
| 117 | "dtype": l.dtype.String(), |
| 118 | "mean": l.mean, |
| 119 | "name": l.name, |
| 120 | "trainable": l.trainable, |
| 121 | "variance": l.variance, |
| 122 | }, |
| 123 | InboundNodes: inboundNodes, |
| 124 | } |
| 125 | } |
| 126 | |
| 127 | func (l *LNormalization) GetCustomLayerDefinition() string { |
| 128 | return `` |