0
當我運行下面的代碼時,變量類型變成了torch.LongTensor。我怎樣才能創建一個torch.cuda.LongTensor而不是?define torch.cuda.LongTensor而不是火炬。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]