interp_test.py 文件源码

def test_left_interp_on_a_vector():
    vector = torch.randn(6)

    res = left_interp(interp_indices, interp_values, Variable(vector)).data
    actual = torch.matmul(interp_matrix, vector)
    assert approx_equal(res, actual)