Commit cc080aba authored by Christian Fuß's avatar Christian Fuß

Added getter method for shape parameter. Fixed a small issue with the order of axes in RNN states

parent 2d98cca1
Pipeline #199119 failed with stages
in 2 minutes and 15 seconds
......@@ -176,6 +176,10 @@ public class ArchitectureElementData {
return getLayerSymbol().getIntValue(AllPredefinedLayers.DIM_NAME).get();
}
public List<Integer> getShape(){
return getLayerSymbol().getIntTupleValue(AllPredefinedLayers.SHAPE_NAME).get();
}
public int getLayers(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.LAYERS_NAME).get();
}
......
......@@ -145,7 +145,7 @@ public abstract class CNNArchTemplateController {
//If the input is an RNN state which is saved as list, take its first entry (unless a state is assigned to another state)
if(input instanceof VariableSymbol && ((VariableSymbol) input).getMember() == VariableSymbol.Member.STATE
&& !(getCurrentElement().getElement() instanceof VariableSymbol && ((VariableSymbol) getCurrentElement().getElement()).getMember() == VariableSymbol.Member.STATE)){
inputNames.add(getName(input) + "[" + 0 + "]");
inputNames.add("F.swapaxes(" + getName(input) + "[" + 0 + "],0,1)");
}else{
inputNames.add(getName(input));
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment