温馨提示×

tensorflow创建张量的方法是什么

小亿
84
2024-03-15 12:38:23
栏目: 深度学习

在TensorFlow中,可以使用tf.constant()函数来创建张量。例如:

import tensorflow as tf

# 创建一个标量
scalar = tf.constant(5)

# 创建一个向量
vector = tf.constant([1, 2, 3])

# 创建一个矩阵
matrix = tf.constant([[1, 2], [3, 4]])

# 创建一个三维张量
tensor = tf.constant([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

0