在手电筒中反转整数类型矩阵时出现错误

奥马尔·阿尔苏威迪

我试图使用3来查找由随机整数组成的3x3矩阵的逆torch.randint()但是,这样做时出现错误:“ Long”未实现“ inverse_cpu”

编码:

A = torch.randint(0, 10, (3, 3))
A_inv = A.inverse()
print(A @ A_inv, "\n", A_inv @ A)

我认为A.inverse()期望矩阵A的逆也可以是整数类型,但事实并非如此。也许我们可以使矩阵A的类型为float torch.Tensor(),或者让A_inv对其进行反转。虽然我不太确定该怎么做。

感谢你的协助!

奥马尔·阿尔苏威迪

好的,我想出了两种方法:

1.) A = torch.randint(0, 10, (3, 3), dtype=torch.float32)

2.) A = torch.Tensor(np.random.randint(0, 10, (3, 3)))

然后反转任何一个都不会产生错误,因为两者现在都是float32类型。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章