如何将[Batch_Size,A,B]张量乘以[Batch_Size,B]张量,得到[Batch_Size,A]张量?

2022-09-28 21:59:15 发布

您现在位置:Python中文网/ 问答频道 /正文

题目中的问题-我如何将这些形状的两个张量相乘,得到那个形状的张量

[Batch_Size, A, B] x [Batch_Size, B] = [Batch_Size, A]

tf.matmul()要求张量具有相同的秩


Tags: sizetfbatch形状题目matmul

热门问题