docs: public static INDArray reshapeCnnMaskToTimeSeriesMask(INDArray timeSeriesMaskAsCnnMask, int minibatchSize) {
          Preconditions.checkArgument(timeSeriesMaskAsCnnMask.rank() == 4 || timeSeriesMaskAsCnnMask.size(1) != 1 ||
                  timeSeriesMaskAsCnnMask.size(2) != 1 || timeSeriesMaskAsCnnMask.size(3) != 1,
                  "Expected rank 4 mask with shape [mb*seqLength, 1, 1, 1]. Got rank %s mask array with shape %s",
                  timeSeriesMaskAsCnnMask.rank(), timeSeriesMaskAsCnnMask.shape());
  
          val timeSeriesLength = timeSeriesMaskAsCnnMask.length() / minibatchSize;
  
          return timeSeriesMaskAsCnnMask.reshape('f', minibatchSize, timeSeriesLength);
      }