TensorFlow – 如何创建一个所有元素都设置为1的张量
TensorFlow是谷歌设计的开源Python库,用于开发机器学习模型和深度学习神经网络。
使用的方法:
- tf.ones:这个方法接受形状和类型,并返回一个给定形状和类型的张量,所有值都设置为1。
- tf.fill:这个方法接受shape、value和type,并返回一个给定的shape和type的张量,所有的值都设置为value。
实例1:本实例使用 ones() 方法创建一个所有元素都设置为1的张量。
# importing the library
import tensorflow as tf
# Generating a Tensor of shape (2, 3)
res = tf.ones(shape = (2, 3))
# Printing the resulting Tensors
print("Res: ", res )
输出:
Res: tf.Tensor(
[[1. 1. 1.]
[1. 1. 1.]], shape=(2, 3), dtype=float32)
实例2:本实例使用fill()方法,其值=1,创建一个所有元素都设置为1的张量。
# importing the library
import tensorflow as tf
# Generating a Tensor of shape (2, 3)
res = tf.fill(dims = (2, 3), value = 1)
# Printing the resulting Tensors
print("Res: ", res )
输出:
Res: tf.Tensor(
[[1 1 1]
[1 1 1]], shape=(2, 3), dtype=int32)