Closed Redmept1on closed 7 months ago
did you test the case on cpu? we test the case on gpu, it reports the same info as torch but the cpu gave [[7,8,9]], we will fix it in the next version
Thanks for your quick response, sorry for wrong information, after verification I found that my code was running on the CPU.
Environment
Hardware Environment(
Ascend
/GPU
/CPU
):device gpu
Software Environment:
Describe the current behavior
[[7. 8. 9.]]
Describe the expected behavior
IndexError: index out of range in self
Steps to reproduce the issue
pytorch v2.0.0+cu117
Special notes for this issue
The index parameter lacks a range check