本文整理汇总了Python中tensorflow.python.ops.nn_ops.data_format_vec_permute函数的典型用法代码示例。如果您正苦于以下问题：Python data_format_vec_permute函数的具体用法？Python data_format_vec_permute怎么用？Python data_format_vec_permute使用的例子？那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。
在下文中一共展示了data_format_vec_permute函数的8个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: testNHWCToNCHW2D
def testNHWCToNCHW2D(self):
x_val = [[7, 4], [9, 3], [4, 5], [5, 1]]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x)
with self.test_session(use_gpu=test_lib.is_gpu_available()) as sess:
y_val = sess.run(y)
self.assertAllEqual(y_val, [[7, 4], [5, 1], [9, 3], [4, 5]])示例2: testNCHWToNHWC2D
def testNCHWToNHWC2D(self):
x_val = [[7, 4], [9, 3], [4, 5], [5, 1]]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x, src_format="NCHW", dst_format="NHWC")
with test_util.use_gpu():
y_val = self.evaluate(y)
self.assertAllEqual(y_val, [[7, 4], [4, 5], [5, 1], [9, 3]])示例3: testNCHWToNHWC
def testNCHWToNHWC(self):
x_val = [7, 4, 9, 3]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x, src_format="NCHW", dst_format="NHWC")
with self.test_session(use_gpu=test_lib.is_gpu_available()) as sess:
y_val = sess.run(y)
self.assertAllEqual(y_val, [7, 9, 3, 4])示例4: testHWNCToNHWC
def testHWNCToNHWC(self):
x_val = [7, 4, 9, 3]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x, src_format="HWNC", dst_format="NHWC")
with test_util.use_gpu():
y_val = self.evaluate(y)
self.assertAllEqual(y_val, [9, 7, 4, 3])示例5: testNHWCToNCHW
def testNHWCToNCHW(self):
x_val = [7, 4, 9, 3]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x)
with test_util.use_gpu():
y_val = self.evaluate(y)
self.assertAllEqual(y_val, [7, 3, 4, 9])示例6: test
def test(self):
x_val = [7, 4, 9, 3]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x)
with self.test_session(use_gpu=test_lib.is_gpu_available()) as sess:
y_val = sess.run(y)
self.assertAllEqual(y_val, [7, 3, 4, 9])示例7: testHWNCToNHWC2D
def testHWNCToNHWC2D(self):
x_val = [[7, 4], [9, 3], [4, 5], [5, 1]]
x = constant_op.constant(x_val)
y = nn_ops.data_format_vec_permute(x, src_format="HWNC", dst_format="NHWC")
with self.session(use_gpu=test_lib.is_gpu_available()) as sess:
y_val = sess.run(y)
self.assertAllEqual(y_val, [[4, 5], [7, 4], [9, 3], [5, 1]])示例8: _runPermuteAndCompare
def _runPermuteAndCompare(self, x, src_format, dst_format, expected):
with self.cached_session() as session:
with self.test_scope():
placeholder = array_ops.placeholder(dtypes.as_dtype(x.dtype), x.shape)
param = {placeholder: x}
output = nn_ops.data_format_vec_permute(
placeholder, src_format=src_format, dst_format=dst_format)
result = session.run(output, param)
self.assertAllEqual(result, expected)