§ x-j´ãó<—ddlmZddlmZmZddlmZ ddd„ZdS)é)Ú annotations)ÚTensorÚ_C_ops)Úin_dynamic_or_pir_modeFÚlhsrÚrhsÚ batch_sizesÚlistÚ trans_lhsÚboolÚ trans_rhsÚreturnú tuple[Tensor]cóR—t¦«rtj|||||¦«SdS)aÀ Cluster launched gemm into one op, which can be further fused and optimized. Args: lhs (Tensor): A tensor shaped in (total_seq_len, input_hidden_size), meant to be perform gemm operation according to batch range. rhs (Tensor): A tensor shaped in (num_batches, input_hidden_size, output_hidden_size). batch_sizes(list): A list of integers representing the number of rows in each batch. trans_lhs (bool): Whether view lhs matrix as last 2D-transposed. Default: False. trans_rhs (bool): Whether view rhs matrix as last 2D-transposed. Default: False. Returns: tuple: - out (Tensor): The result of batched gemm operation. N)rrÚ batched_gemm)rrr r r s új/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddle/incubate/nn/functional/batched_gemm.pyrrs8€õ,ÑÔðPÝÔ" 3¨¨[¸)ÀYÑOÔOÐOðPðPóN)FF) rrrrr r r r r r rr)Ú __future__rÚpaddlerrÚpaddle.frameworkrr©rrúrsuðð#Ð"Ð"Ð"Ð"Ð"à!Ð!Ð!Ð!Ð!Ð!Ð!Ð!Ø3Ð3Ð3Ð3Ð3Ð3ðØð PðPðPðPðPðPðPr