【问题标题】:Inner product of Tensors张量的内积
【发布时间】:2021-01-25 04:01:03
【问题描述】:

谁能解释一下如何在 python 中做两个张量的内积以获得一维数组。例如,我有两个大小为 (6,6,6,6,6) 和 (6,6,6,6) 的张量。我需要一个大小为 (6,1) 或 (1,6) 的一维数组。

【问题讨论】:

    标签: python-3.x multidimensional-array tensor dot-product numpy-einsum


    【解决方案1】:

    Numpy 有一个函数tensordot,查看一下:
    https://numpy.org/doc/stable/reference/generated/numpy.tensordot.html

    【讨论】:

      猜你喜欢
      • 2019-06-24
      • 1970-01-01
      • 1970-01-01
      • 2020-03-21
      • 1970-01-01
      • 1970-01-01
      • 2018-03-14
      • 2019-01-07
      • 2017-06-14
      相关资源
      最近更新 更多