1 报错描述
1.1 系统环境
Hardware Environment(Ascend/GPU/CPU): GPU
Software Environment:
– MindSpore version (source or binary): 1.6.0
– Python version (e.g., Python 3.7.5): 3.7.6
– OS platform and distribution (e.g., Linux Ubuntu 16.04): Ubuntu 4.15.0-74-generic
– GCC/Compiler version (if compiled from source):
1.2 基本信息
1.2.1 脚本
根据标杆torch算子NLLLoss的用例,编写输入为(N, C, d1, d2)的用例,脚本如下:
01 loss = nn.NLLLoss()
02 input = torch.randn(5, 4, 8, 8)
03 m = nn.LogSoftmax(dim=1)
04 target = torch.empty(5, 8, 8, dtype=torch.long).random