addmm_ is deprecated: addmm_(Number beta, Number alpha, Tensor mat1, Tensor mat2) Consider using one of the following signatures instead: addmm_(Tensor mat1, Tensor mat2, *, Number beta, Number alpha) (Triggered internally at /pytorch/torch/csrc/utils/python_arg_parser.cpp:1005.) distmat.addmm_(1, -2, x, self.centers.t())
解决方案:将
distmat.addmm_(1, -2, x, self.centers.t())
改为
distmat.addmm_(x, self.centers.t(), beta=1, alpha=-2)
问题即可解决!
版权声明:本文为power_kaikaige原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。