Tensorflow js, getting error while using matMul

Question

let tensor1 = tf.tensor(m1);
let tensor2 = tf.tensor(m2);
console.log(tensor1.shape, tensor2.shape)
result = tf.matMul(tensor1, tensor2)

The output of shapes are => [ 6890, 3, 10 ] [ 10 ]

While trying to use matMul this is the error => Error: Error in matMul: inner shapes (10) and (undefined) of Tensors with shapes 6890,3,10 and 10 and transposeA=false and transposeB=false must match.

The same multiplication works fine in python using numpy's matmul.

AloneTogether · Accepted Answer · 2022-09-07 07:20:11Z

1

Maybe something like this:

let tensor1 = tf.randomNormal([6890, 3, 10]);
let tensor2 = tf.randomNormal([10]).expandDims(-1);
result = tf.matMul(tensor1, tensor2).squeeze(-1)
console.log(result.shape)

[6890,3]

answered Sep 7, 2022 at 7:20

AloneTogether

26.8k5 gold badges23 silver badges43 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Tensorflow js, getting error while using matMul

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related