张量流–如何创建所有元素都设置为一的张量
原文:https://www . geesforgeks . org/tensorflow-如何创建一个所有元素集为一的张量/
TensorFlow 是谷歌设计的开源 Python 库，用于开发机器学习模型和深度学习神经网络。
使用的方法:
- [T0】 TF. one: 【T1] This method accepts shapes and types and returns tensors of given shapes and types, all of which are set to 1.
- TF. fill: This method accepts shapes, values and types, and returns tensors of given shapes and types, all values are set to values.
示例 1: 本示例使用 ones()方法创建所有元素都设置为 1 的张量。
Python 3
# 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 的张量。
Python 3
# 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)
版权属于：月萌API www.moonapi.com，转载请注明出处
