Crate caffe2op_matmul

source ·

Structs

  • GetMatMulGradient
  • MatMulOp
    | Matrix multiplication $Y = A * B$, where | A has size (M x K), B has size (K x N), | and Y will have a size (M x N). | | To transpose A or B before multiplication, | pass 1 to the trans_a and/or trans_b | arguments, which separate the first | and second dimensions of the respective | matrices using axis_a and axis_b. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/matmul_op.cc |