Pytorch テンソルから NumPy 配列へ質問する

Question

も使用する必要があると思います.detach()。私は、CUDA と GPU を使用する Colab で Tensor を numpy 配列に変換する必要がありました。次のようにしました。

# this is just my embedding matrix which is a Torch tensor object
embedding = learn.model.u_weight

embedding_list = list(range(0, 64382))

input = torch.cuda.LongTensor(embedding_list)
tensor_array = embedding(input)
# the output of the line below is a numpy array
tensor_array.cpu().detach().numpy()

Answer 1

も使用する必要があると思います.detach()。私は、CUDA と GPU を使用する Colab で Tensor を numpy 配列に変換する必要がありました。次のようにしました。

# this is just my embedding matrix which is a Torch tensor object
embedding = learn.model.u_weight

embedding_list = list(range(0, 64382))

input = torch.cuda.LongTensor(embedding_list)
tensor_array = embedding(input)
# the output of the line below is a numpy array
tensor_array.cpu().detach().numpy()

Pytorch テンソルから NumPy 配列へ質問する

ベストアンサー1

おすすめ記事