tf.zeros(), tf.zeros_like(), tf.ones(),tf.ones_like()

简介: 【8月更文挑战第11天】tf.zeros(), tf.zeros_like(), tf.ones(),tf.ones_like()。

tf.zeros(), tf.zeros_like(), tf.ones(),tf.ones_like()
因为tf.ones(),tf.ones_like()与tf.zeros(),tf.zeros_like()的用法相似,因此下面只演示前者的使用方法。
创建一个值为0的常量。
tf.zeros(shape, dtype=tf.float32, name=None):
shape:张量形状;
dtype:类型;
name:名称。

代码:
zeros_b = tf.zeros(shape=[2, 3], dtype=tf.int32) # 创建2x3矩阵,元素值均为0

根据输入张量创建一个值为0的张量,形状和输入张量相同。
tf.zeros_like(input_tensor, dtype=None, name=None, optimize=True):
input_tensor:张量;
dtype:类型;
name:名称;
optimize:优化。

代码:
zeros_like_c = tf.zeros_like(const_a)

查看生成数据

zeros_like_c.numpy()
输出:
array([[0., 0.],
[0., 0.]], dtype=float32)

相关文章
|
4月前
|
机器学习/深度学习 Python
tf.keras.layers.LSTM/tf.keras.layers.LSTMCell
【8月更文挑战第20天】tf.keras.layers.LSTM/tf.keras.layers.LSTMCell。
45 1
|
4月前
tf.random
【8月更文挑战第12天】tf.random。
46 3
|
2月前
|
TensorFlow 算法框架/工具
Tensorflow error(二):x and y must have the same dtype, got tf.float32 != tf.int32
本文讨论了TensorFlow中的一个常见错误,即在计算过程中,变量的数据类型(dtype)不一致导致的错误,并通过使用`tf.cast`函数来解决这个问题。
26 0
|
4月前
tf.keras.layers.Dense
【8月更文挑战第20天】tf.keras.layers.Dense。
81 2
|
4月前
|
算法框架/工具 数据格式
tf.keras.layers.Conv2D
【8月更文挑战第20天】tf.keras.layers.Conv2D。
53 2
|
4月前
tf.keras.layers.MaxPooling2D/AveragePooling2D
【8月更文挑战第19天】tf.keras.layers.MaxPooling2D/AveragePooling2D。
49 0
torch中如何将tensor([[1, 2, 3]]) 和 tensor([4]) 合并成 tensor([[1,2,3,4]])
可以使用 torch.cat() 方法将两个张量沿着指定的维度进行拼接
493 0
|
定位技术 内存技术
TF 卡是什么
TF 卡是什么
491 0
下一篇
DataWorks