weixin_42038795的博客

原创 K.batch_dot()的用法总结

K.batch_dot()的用法总结 K.batch_dot(a, b, axis) 输入1：a(计算量1，维度应该大于等于2) 输入2：b(计算量2，维度应该大于等于2) 输入3：axis(可以是一个元组(x,y)，也可以是一个整数) 当axis是一个元组(x,y)时，x代表a参与计算的维数，y代表b参与计算的维度； 当axis是一个整数z时，则a和b参与计算的维数均为z； 当axis为none时，则a参与计算的维数是倒数第一维，b参与计算的维数是倒数第二维(听说有的情况是b也是倒数第一维，比较少遇到)。