torch.Tensor манипуляция - вызов методов на субвекторах

У меня есть тензор, который представляет собой вектор с 9 ячейками, я хотел бы вытащить ячейки с 3 по 6 без цикла, как в Matlab.

Больше информации:

th> my_table = {0.1, 0.2, 0.3, ... 0.9}
-- I create tensor out of it.
th> t_table = torch.Tensor(my_table )
-- I get the vector 0.1, 0.2, 0.3, ..., 0.9
-- Now I will get it's mean like this:
th> torch.mean(t_table)
0.5
-- Now I would like to do for example torch.mean on cells 3-6.
th> --<what to write here?>

person oria general    schedule 23.11.2015    source источник


Ответы (1)


У меня нет опыта работы с библиотекой torch, но документация для _1 _ показывает, что существует :sub и :select < / a> мета-методы, а также отличный _ 4_, который поддерживает нарезку тензора.

Вы можете использовать один из них, чтобы получить тензор для индексов от 3 до 6 и применить к нему torch.mean.

person hjpotter92    schedule 23.11.2015