Интересно, как я могу урезать точность тензоров на GPU до желаемых знаков после запятой? Я хочу ограничить элементы в тензоре 4 знаками после запятой, но кажется, что в Torch нет встроенной функции, чтобы сделать это эффективно и действенно. Кто-нибудь знает, как это можно сделать?
Как ограничить/округлить элементы тензора до 4 знаков после запятой на GPU в Torch?
Ответы (1)
По-прежнему непросто округлять числа в тензоре, но простое решение (которое подходит для моих целей) состоит в том, чтобы просто преобразовать тензор в половинную точность следующим образом:
require 'cutorch'
temp = torch.rand(3, 3)
temp = temp:cuda()
temp = temp:cudaHalf()
person
Amir
schedule
15.03.2017