【发布时间】:2015-09-21 04:06:28
【问题描述】:
我正在使用 Torch7 来训练一些神经网络,我有一个昏暗 1(向量)的张量,我想访问这个向量中的元素 i。 不幸的是,它给了我一个整数而不是一个大小为 1 的张量。
我明白了:
matrix = torch.Tensor{1,2}
> 1
2
[torch.DoubleTensor of size 2]
matrix[1]
> 1
我想要这个:
matrix[1]
> 1
[torch.DoubleTensor of size 1]
我有义务这样做:
torch.Tensor{matrix[1]}
> 1
[torch.DoubleTensor of size 1]
【问题讨论】:
标签: torch