Method constructor

tfjs-layers/src/engine/dataset_fakes.ts:98–113

(args: FakeDatasetArgs)

  private tensorIndex = 0;

  constructor(args: FakeDatasetArgs) {
    super();
    this.xBatchShape = mergeBatchSizeAndShape(args.batchSize, args.xShape);
    this.yBatchShape = mergeBatchSizeAndShape(args.batchSize, args.yShape);
    this.numBatches = args.numBatches;
    this.batchCount = 0;
    this.xTensorsFunc = args.xTensorsFunc;
    this.yTensorsFunc = args.yTensorsFunc;

    // Sanity check on the preset tensors.
    tfc.util.assert(
        this.xTensorsFunc == null && this.yTensorsFunc == null ||
            this.xTensorsFunc != null && this.yTensorsFunc != null,
        () => 'presetXTensors and presetYTensors must be both null/undefined ' +
            'or both set.');
  }

  async next(): Promise<IteratorResult<FitDatasetElement>> {
    const done = ++this.batchCount > this.numBatches;