torch.LongTensorの代わりにtorch.cuda.LongTensorを定義します

Question

次のコードを実行すると、変数の型がtorch.LongTensorに変わります。代わりにどのようにtorch.cuda.LongTensorを作成することができますか？

# Turn string into list of longs 
def char_tensor(string): 
    tensor = torch.zeros(len(string)).long() 
    for c in range(len(string)):   
     tensor[c] = all_characters.index(string[c]) 
    return Variable(tensor) 

print(char_tensor('abcDEF')) 

出力：

Variable containing: 
10 
11 
12 
39 
40 
41 
[torch.LongTensor of size 6] 

Answer 1

正解：

# Turn string into list of longs 
def char_tensor(string): 
    tensor = torch.zeros(len(string)).long() 
    for c in range(len(string)):   
     tensor[c] = all_characters.index(string[c]) 
    return Variable(tensor).cuda() 

print(char_tensor('abcDEF'))