Output tensor shape: 1, outputDimension (typically 768)
1, outputDimension
final int outputDimension;