如何在TensorFlow中創(chuàng)建一個(gè)張量

小樊
86
2024-03-01 19:24:14

在 TensorFlow 中，可以使用 tf.constant() 函數(shù)創(chuàng)建一個(gè)張量。示例如下：

import tensorflow as tf

# 創(chuàng)建一個(gè)常量張量
tensor = tf.constant([1, 2, 3])

# 創(chuàng)建一個(gè)全零張量
zero_tensor = tf.zeros([2, 3])

# 創(chuàng)建一個(gè)全一張量
ones_tensor = tf.ones([2, 3])

以上代碼分別創(chuàng)建了一個(gè)常量張量、一個(gè)全零張量和一個(gè)全一張量?？梢愿鶕?jù)需要調(diào)整張量的形狀和數(shù)值。TensorFlow 還提供了其他創(chuàng)建張量的函數(shù)，如 tf.Variable()、tf.zeros_like()、tf.ones_like() 等。

0 踩