• L
    fix expand op: (#19302) · e9e3c087
    liym27 提交于
    1. add tensor support for argument expand_times in expand op;
    2. add support parameter inference when argument expand_times is a list containing integer and tensor variable;
    
    improve expand op according to reviews:
    1. add doc of ExpandTimes in expand_op.cc;
    2. improve the test of test_api.
    
    add stop_gradient=True when attr(expand_times) is tensor Variable, change code examples.
    test=develop,test=document_preview
    e9e3c087
expand_op.h 8.7 KB