docs: @Override
      public INDArray rand(int[] shape, float min, float max, org.nd4j.linalg.api.rng.Random rng) {
          //ensure shapes that wind up being scalar end up with the write shape
          if (shape.length == 1 && shape[0] == 0) {
              shape = new int[] {1, 1};
          }
          return Nd4j.getDistributions().createUniform(min, max).sample(shape);
      }