Сохраните bfloat16 в двоичном формате. ⇐ Python

[Anonymous]

Каков идиоматический способ сохранения bfloat torch.tensor на диск в виде необработанного двоичного файла? Код ниже выдаст ошибку, поскольку numpy не поддерживает bfloat16.
import torch
import numpy as np

tensor = torch.tensor([1, 2, 3, 4, 5]).bfloat16()

# TypeError: Got unsupported ScalarType BFloat16
arr = tensor.numpy()
arr.tofile("output.bin")


Подробнее здесь: https://stackoverflow.com/questions/767 ... ary-format