【问题标题】:Subtract a vector across all columns in a tensor for Torch在 Torch 的张量中减去所有列的向量
【发布时间】:2016-02-02 05:05:53
【问题描述】:

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

【问题讨论】:

  • 您能提供您已经尝试过的示例代码吗?请参阅stackoverflow.com/help/mcve 以获得帮助编写一个可以得到答案的问题

标签: lua torch


【解决方案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]

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2016-03-17
    • 2016-07-02
    • 1970-01-01
    • 2021-01-14
    • 2023-04-02
    • 1970-01-01
    • 2021-10-07
    • 2019-04-20
    相关资源
    最近更新 更多