trace_test.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:gpytorch 作者: jrg365 项目源码 文件源码
def test_trace_components_normal_matrices():
    a_mat = torch.randn(3, 4)
    b_mat = torch.randn(3, 4)

    a_res, b_res = trace_components(a_mat, b_mat)
    assert torch.equal(a_res, a_mat)
    assert torch.equal(b_res, b_mat)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号