Function generateRandomTensorContainer

tfjs-layers/src/engine/dataset_fakes.ts:73–85

(shape: Shape|{[name: string]: Shape})

}

function generateRandomTensorContainer(shape: Shape|{[name: string]: Shape}):
    tfc.Tensor|{[name: string]: tfc.Tensor} {
  let output: tfc.Tensor|{[name: string]: tfc.Tensor};
  if (Array.isArray(shape)) {
    output = tfc.randomNormal(shape);
  } else {
    output = {};
    for (const name in shape) {
      output[name] = tfc.randomNormal(shape[name]);
    }
  }
  return output;
}

class FakeNumericIterator extends LazyIterator<FitDatasetElement> {
  private xBatchShape: Shape|{[name: string]: Shape};