def test_sym_toeplitz_constructs_tensor_from_vector():
c = torch.Tensor([1, 6, 4, 5])
res = utils.toeplitz.sym_toeplitz(c)
actual = torch.Tensor([
[1, 6, 4, 5],
[6, 1, 6, 4],
[4, 6, 1, 6],
[5, 4, 6, 1],
])
assert torch.equal(res, actual)
评论列表
文章目录