code: @JsonIgnore
      public Map<String, long[]> getParamShapes() {
          Map<String, long[]> map = new LinkedHashMap<>();
          map.putAll(weightParams);
          map.putAll(biasParams);
          return map;
      }
docs: Get the parameter shapes for all parameters
  
  @return Map of parameter shapes, by parameter