Method addMatrixMultiply

include/NvInfer.h:7429–7433

\brief Add a MatrixMultiply layer to the network. \param input0 The first input tensor (commonly A). \param op0 The operation to apply to input0. \param input1 The second input tensor (commonly B). \param op1 The operation to apply to input1. The inputs are shape tensors if the output is a shape tensor. \see IMatrixMultiplyLayer \warning Int32 tensors are not valid input tensors. \return The

    //! \return The new matrix multiply layer, or nullptr if it could not be created.
    //!
    IMatrixMultiplyLayer* addMatrixMultiply(
        ITensor& input0, MatrixOperation op0, ITensor& input1, MatrixOperation op1) noexcept
    {
        return mImpl->addMatrixMultiply(input0, op0, input1, op1);
    }

    //!
    //! \brief Add a nonzero layer to the network.