Преобразование тензоров Pytorch Bfloat16 в Numpy Throws Typeerror ⇐ Python

[Anonymous]

Когда вы пытаетесь преобразовать тензор Forch Bfloat16 в массив Numpy, он бросает TypeError :

Код: Выделить всё

import torch

x = torch.Tensor([0]).to(torch.bfloat16)
x.numpy()  # TypeError: Got unsupported ScalarType BFloat16

import numpy as np
np.array(x)  # same error

Есть ли обходная оболочка, чтобы сделать это преобразование?


Подробнее здесь: https://stackoverflow.com/questions/781 ... -typeerror