Lines Matching refs:reshape
40 xr = x.reshape([num, x.shape[-2], x.shape[-1]])
41 yr = y.reshape([num, y.shape[-2], y.shape[-1]])
42 zr = z.reshape([num, z.shape[-2], z.shape[-1]])
55 x = np.array([0., 1., 2., 3.]).reshape([1, 2, 2])
56 y = np.array([1., 2., 3., 4.]).reshape([1, 2, 2])
58 z1 = np.array([3., 4., 11., 16.]).reshape([1, 2, 2])
61 x = np.array([1., (1j), (-1.), (-1j)]).reshape([1, 2, 2])
64 z1 = np.array([2., (2.j), -2., (-2.j)]).reshape([1, 2, 2])
68 z1 = np.array([(2. - 2.j), (-2. + 2.j), (-2. + 2.j), (2. - 2.j)]).reshape(
73 z1 = np.array([(2. + 2.j), (-2. + 2.j), (2. - 2.j), (2. + 2.j)]).reshape(
82 x = x_in if not adjoint_a else x_in.reshape(x_t_shape)
83 y = y_in if not adjoint_b else y_in.reshape(y_t_shape)
104 return vals.reshape(shape)
153 x = x_in if not adjoint_a else x_in.reshape(x_t_shape)
154 y = y_in if not adjoint_b else y_in.reshape(y_t_shape)
178 x = np.random.normal(0, 1, b * n * k).astype(dtype).reshape([b, n, k])
181 b * n * k).astype(dtype).reshape([b, n, k])
182 y = np.random.normal(0, 1, b * k * m).astype(dtype).reshape([b, k, m])
185 b * k * m).astype(dtype).reshape([b, k, m])
