output_node is a list containing some output nodes name. Now I solved the infer shape problem by set attr['size']'s value to (10,10) in ...
確定! 回上一頁