Function test_make_vector

UTest/correlation_test.py:9–16

()

def test_make_vector():
  x = tf.constant([[
    [[1, 1.1, 1.2], [2, 2.1, 2.2], [3, 3.1, 3.2]],
    [[4, 4.1, 4.2], [5, 5.1, 5.2], [6, 6.1, 6.2]],
    [[7, 7.1, 7.2], [8, 8.1, 8.2], [9, 9.1, 9.2]]
  ]], 'float32')
  y = _make_vector(x, 3)
  return y


def test_make_displacement():