Lines Matching refs:numpy
21 import numpy
34 input_tensor = numpy.arange(12).reshape((3, 4))
35 desired_shape = numpy.array([6, None])
45 input_tensor = numpy.arange(12).reshape((3, 4))
46 desired_shape = numpy.array([5, None])
56 input_tensor = numpy.arange(2 * 2 * 4).reshape((2, 2, 4))
57 desired_shape = numpy.array([4, 4, None])
58 output_tensor = numpy.array([[[0], [2], [4], [6]], [[1], [3], [5], [7]],
74 input_tensor = numpy.arange(2 * 2 * 2 * 8).reshape((2, 2, 2, 8))
75 desired_shape = numpy.array([4, 4, 4, None])
76 output_tensor = numpy.array(
94 input_tensor = numpy.zeros(shape=[1, 2, 2, 4])
