AlphaTensor 上的维度不匹配
作者: ywsslr创建于 2024年7月10日更新于 2024年7月21日
最近我想要基于 AlphaZero 实现 AlphaTensor。说实话，该仓库及其材料 https://www.kdnuggets.com/2023/03/first-open-source-implementation-deepmind-alphatensor.html 非常有帮助。但是当我想使用 2*2 矩阵的乘法来训练模型时，程序会向我报告不匹配的错误。以下代码段代表了我的配置：
内容来源: nebuly-ai/optimate