我有一个张量T的形状Batch_Size x Num_Items x Item_Dimension和另一个张量P的形状Batch_Size x Num_Items,其中Num_Items值在每批P之和为根据概率分布P,我想在没有替换N项的情况下从T中取样本。由此得到的张量应该是Batch_Size x N x Item_Dimension形状的。我该怎么做?
我对张量代数不太熟悉,我很难理解如何让numpy.tensordot做我想做的事。我使用的示例很简单:给定一个带有形状(2,2,3)的张量(2,2,3)和另一个具有形状(2,1,3)的b,我想要一个具有shape (2,1)的结果张量c。此张量将是以下等效python代码的结果:c = np.zeros((2,n)) c