Каков идиоматический способ сохранения 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
