Pytorch - Tensor변수에서 값만 가져오기 ( item() 함수 )

Pytorch에서 tensor에 저장된 값만을 가져오는 방법.

 

tensor변수에 .item() 을 붙이면 된다.

예)

 ten_val = torch.one()

 val = ten_val.item() # 텐서에서 값만을 가져온다.

item() 적용