Python transposeの例

コード例 #1

 def testTranspose(self):
   for dtype in self.numeric_types:
     v = np.arange(4, dtype=np.int32).astype(dtype).reshape([2, 2])
     self._assertOpOutputMatchesExpected(
         lambda x: xla.transpose(x, [1, 0]), args=(v,), expected=v.T)

コード例 #2

ファイル: xla_ops_test.py プロジェクト: Ajaycs99/tensorflow

 def testTranspose(self):
   for dtype in self.numeric_types:
     v = np.arange(4, dtype=np.int32).astype(dtype).reshape([2, 2])
     self._assertOpOutputMatchesExpected(
         lambda x: xla.transpose(x, [1, 0]), args=(v,), expected=v.T)