hw2 q3 Adam的memory测试无法·通过

抱歉打扰，我在做24fall的版本，本来自己写的过不了，参考后发现要在结果转换为tensor，但是memory测试过不了，说明有可以用.data的地方没有用但排查没有想到哪里可以，想问问你能不能通过memory test，另外看你也在看llm推理方向，可不可以交流一下hh