代码之家  ›  专栏  ›  技术社区  ›  katiex7

对于火炬张量,和和和和有什么区别?

  •  0
  • katiex7  · 技术社区  · 5 年前

    我在使用python sum或torch.sum时得到了相同的结果,那么为什么torch要实现一个sum函数呢?他们之间有区别吗?

    0 回复  |  直到 5 年前
        1
  •  0
  •   Separius    5 年前

    没有什么, torch.sum 电话 tensor.sum 和蟒蛇的 sum 电话 __add__ (或) __radd__ 如果需要的话)哪个电话 求和 再一次

    所以唯一的区别是函数调用的数量,以及 tensor.sum() 应该是最快的(当你有小张量并且函数调用的开销很大时)