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

Question

给定尺寸NxM和向量Nx1的张量，如何减去Torch中张量的每列中的向量？

Answer 1

一种可能性是使用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) -- make a 3x1 tensor before expand
print(C)
--  0  1
--  2  3
--  4  5
-- [torch.DoubleTensor of size 3x2]