ホームページ > バックエンド開発 > Python チュートリアル > TensorFlow テンソルを NumPy 配列に変換するにはどうすればよいですか?

TensorFlow テンソルを NumPy 配列に変換するにはどうすればよいですか?

Mary-Kate Olsen
リリース: 2024-11-03 17:54:30
オリジナル
782 人が閲覧しました

How to Convert TensorFlow Tensors to NumPy Arrays?

TensorFlow でテンソルを NumPy 配列に変換する方法

TensorFlow の Python バインディングでは、テンソルを NumPy 配列に変換することは、さらなる目的のために必要なステップですデータ操作またはサードパーティ ライブラリとの統合。

TensorFlow 2.x:

TensorFlow 2.x では、デフォルトで積極的な実行が有効になっており、単に を呼び出すことができます。 Tensor オブジェクトの numpy()。このメソッドは NumPy 配列を返します:

<code class="python">import tensorflow as tf

a = tf.constant([[1, 2], [3, 4]])
b = tf.add(a, 1)

a.numpy()  # [array([[1, 2], [3, 4]], dtype=int32)]
b.numpy()  # [array([[2, 3], [4, 5]], dtype=int32)]</code>
ログイン後にコピー

TensorFlow 1.x の場合:

Eager Execution はデフォルトでは有効になっていません。 TensorFlow 1.x でテンソルを NumPy 配列に変換するには:

  • セッション内で .eval() メソッドを使用します:
<code class="python">a = tf.constant([[1, 2], [3, 4]])
b = tf.add(a, 1)

with tf.Session() as sess:
    out = sess.run([a, b])
    # out[0] contains the NumPy array representation of a
    # out[1] contains the NumPy array representation of b</code>
ログイン後にコピー
  • Use tf .compat.v1.numpy_function:
<code class="python">a = tf.constant([[1, 2], [3, 4]])
b = tf.add(a, 1)

out = tf.compat.v1.numpy_function(lambda x: x.numpy(), [a, b])
# out[0] contains the NumPy array representation of a
# out[1] contains the NumPy array representation of b</code>
ログイン後にコピー

注: NumPy 配列は Tensor オブジェクトとメモリを共有する場合があります。一方への変更はもう一方にも反映される可能性があります。

以上がTensorFlow テンソルを NumPy 配列に変換するにはどうすればよいですか?の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。

ソース:php.cn
このウェブサイトの声明
この記事の内容はネチズンが自主的に寄稿したものであり、著作権は原著者に帰属します。このサイトは、それに相当する法的責任を負いません。盗作または侵害の疑いのあるコンテンツを見つけた場合は、admin@php.cn までご連絡ください。
著者別の最新記事
人気のチュートリアル
詳細>
最新のダウンロード
詳細>
ウェブエフェクト
公式サイト
サイト素材
フロントエンドテンプレート