Hi there,

I use bmm to multiply batches of matrices:

data = Variable(FloatTensor(BATCHSIZE,42,100).normal_(std=1)) # BATCH x 42 x 100

A = Variable(FloatTensor(BATCHSIZE,100,100).normal_(std=1))

B = bmm(data,A) # BATCH x 42 x 100 (OK)

Now I’d like to batch-multiply data with *multiple* matrices:

A = Variable(FloatTensor(BATCHSIZE,NLAYERS,100,100).normal_(std=1))

B = bmm(data,A) # doesn’t work

bmm() gives me the following error:

RuntimeError: invalid argument 2: expected 3D tensor, got 4D

Is there a way to batch-multiple matrices (data) with *3D*-Tensors (A) ?

Thanks in advance!