对 torch.Tensor 进行操作 - 在子向量上调用方法

我有一个张量,它是带有 9 个元素的向量,我想在不循环的情况下提取第 3 到 6 个元素,就像在 Matlab 中一样。

更多信息:

th> my_table = {0.1, 0.2, 0.3, ... 0.9}
-- 我从中创建了张量。
th> t_table = torch.Tensor(my_table )
-- 我获得了向量 0.1、0.2、0.3、...、0.9。
-- 现在我将这个向量的均值计算如下:
th> torch.mean(t_table)
0.5
-- 现在我想执行例如在第 3-6 个单元格上执行 torch.mean。
th>--<在这里写什么?>
点赞
用户1190388
用户1190388

我对torch库没有经验,但是torch.Tensor 的文档显示,存在一个 :sub 和一个 :select元方法以及一个非常好的 __index方法,支持对张量进行切片。

您可以使用其中之一来获取索引 3 到 6 的张量,并在其上应用 torch.mean

2015-11-23 13:54:37