Unfortunately, Tensorflow doesn’t have docker builds for arm64, only for amd64. But you can use third-party build from another vendors.
I found builds from www.arm.com
So you can just pull Tensorflow image for arm64:
docker pull armswdev/tensorflow-arm-neoverse:latest
and run with mount option (-v)
docker run -it -v /your_path/:/your_path_in_docker armswdev/tensorflow-arm-neoverse:latest bash