[python] 'torch.dtype' object has no attribute 'type'

cateto·2021년 11월 5일
0

import numpy as np
... 생략

cos = util.pytorch_cos_sim(embedding_a, embedding_b)[0]
doc_cohesion.append(cos.item())
cos_sim = np.mean(doc_cohesion)

이 상황에서 cosine similarity의 값은 tensor type으로 떨어진다.
그래서 cos.item() 을 활용하여 float형으로 떨어지도록 해야함.

출처 : https://discuss.pytorch.org/t/fastest-way-to-convert-tensor-to-float/21282

profile
Curious for Everything

0개의 댓글