在 Torch 中为张量中的所有列减去向量

2016-2-1 16:14:6
收藏：0
阅读：43
评论：1

给定一个维度为 NxM 的张量和一个 Nx1 的向量，在 Torch 中如何在张量的每列中减去向量？

点赞
用户1688185
用户1688185

一种可能的方法是使用 expand。例如：

local A = torch.Tensor{{1, 2},{3, 4},{5,6}}
local B = torch.ones(3)
local C = A - B:view(3, 1):expandAs(A) -- 在 expand 之前制作一个 3x1 的张量
print(C)
--  0  1
--  2  3
--  4  5
-- [torch.DoubleTensor of size 3x2]
2016-02-01 21:05:53

作者:

用户1879926

用户1879926
推荐文章