Method call

tfjs-layers/src/layers/preprocessing/random_height.ts:115–143

(inputs: Tensor<Rank.R3>|Tensor<Rank.R4>,
    kwargs: Kwargs)

  }

  override call(inputs: Tensor<Rank.R3>|Tensor<Rank.R4>,
    kwargs: Kwargs): Tensor[]|Tensor {

    return tidy(() => {
      const input = getExactlyOneTensor(inputs);
      this.imgWidth = input.shape[input.shape.length - 2];
      const imgHeight = input.shape[input.shape.length - 3];

      this.heightFactor = randomUniform([1],
        (1.0 + this.heightLower), (1.0 + this.heightUpper),
        'float32', this.randomGenerator.next()
      );

      let adjustedHeight = this.heightFactor.dataSync()[0] * imgHeight;
      adjustedHeight = Math.round(adjustedHeight);

      const size:[number, number] = [adjustedHeight, this.imgWidth];

      switch (this.interpolation) {
        case 'bilinear':
          return image.resizeBilinear(inputs, size);
        case 'nearest':
          return image.resizeNearestNeighbor(inputs, size);
        default:
          throw new Error(`Interpolation is ${this.interpolation}
          but only ${[...INTERPOLATION_METHODS]} are supported`);
      }
    });
  }
}

serialization.registerClass(RandomHeight);