Remove some `tf.Variables` to allow easier use of `tf.function`.
Remove some `tf.Variables` to allow easier use of `tf.function`.

Add a simpler-octave loop, before getting into the complexity of tiling.

PiperOrigin-RevId: 263587233

tensorflow/docs,tensorflow/docs,tensorflow/docs