Fix dump of bfloat16 torch tensor

close https://github.com/huggingface/datasets/issues/7000