相关疑难解决方法(0)

tensorflow batch_matmul如何工作?

Tensorflow有一个名为batch_matmul的函数,它可以将更高维的张量相乘.但是我很难理解它是如何工作的,也许部分是因为我很难想象它.

在此输入图像描述

我想要做的是将矩阵乘以3D张量的每个切片,但我不太明白张量a的形状是什么.z是最里面的维度吗?以下哪项是正确的?

在此输入图像描述

我最希望第一个是正确的 - 它对我来说最直观,很容易在.eval()输出中看到.但我怀疑第二个是正确的.

Tensorflow说batch_matmul执行:

out[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])
Run Code Online (Sandbox Code Playgroud)

那是什么意思?在我的例子中,这意味着什么?什么与什么相乘?为什么我没有按照预期的方式获得3D张量?

numpy scipy tensorflow

11
推荐指数
2
解决办法
1万
查看次数

标签 统计

numpy ×1

scipy ×1

tensorflow ×1