使用softmax计算大矩阵报错

[Wenze7]

请提出你的问题 Please ask your question

我的矩阵是25638450000的，在最后一维使用softmax进行降维，报了下面的错 怎么能让我计算这个大矩阵呢，设备是A100 80G

[paddle-bot[bot]]

您好，我们已经收到了您的问题，会安排技术人员尽快解答您的问题，请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时，您也可以通过查看官网API文档、常见问题、历史Issue、AI社区来寻求解答。祝您生活愉快～

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the API，FAQ，Github Issue and AI community to get the answer.Have a nice day!

[yt605155624]

可以先在 issue 区搜下 CUDNN_STATUS_NOT_SUPPORTED 尝试可能的解决办法

[w5688414]

可以尝试切分数据分片进行softmax