【发布时间】:2016-02-02 05:05:53
【问题描述】:
给定一个维度为 NxM 的张量和一个向量 Nx1,如何在 Torch 中减去该张量的每一列中的向量?
【问题讨论】:
-
您能提供您已经尝试过的示例代码吗?请参阅stackoverflow.com/help/mcve 以获得帮助编写一个可以得到答案的问题
给定一个维度为 NxM 的张量和一个向量 Nx1,如何在 Torch 中减去该张量的每一列中的向量?
【问题讨论】:
一种可能性是使用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]
【讨论】: