[Docs] Align code lines of TensorFlow types with flytesnacks

[JiangJiaWei1103]

Why are the changes needed?

The code lines of Tensorflow types are out of sync of the Flytesnacks example.

What changes were proposed in this pull request?

Fix code lines to align with the code example in flytesnacks.

Docs link

https://flyte--5983.org.readthedocs.build/en/5983/user_guide/data_types_and_io/tensorflow_type.html

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 36.90%. Comparing base (25cfe16) to head (2f72dd3).
Report is 6 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #5983      +/-   ##
==========================================
+ Coverage   33.09%   36.90%   +3.80%     
==========================================
  Files        1013     1310     +297     
  Lines      107489   131372   +23883     
==========================================
+ Hits        35570    48477   +12907     
- Misses      68767    78682    +9915     
- Partials     3152     4213    +1061     

Flag	Coverage Δ
unittests-datacatalog	51.58% <ø> (ø)
unittests-flyteadmin	54.05% <ø> (?)
unittests-flytecopilot	22.23% <ø> (ø)
unittests-flytectl	62.39% <ø> (-0.05%)	⬇️
unittests-flyteidl	6.92% <ø> (ø)
unittests-flyteplugins	53.84% <ø> (ø)
unittests-flytepropeller	42.90% <ø> (ø)
unittests-flytestdlib	55.31% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[davidmirror-ops]

Thank you!