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)

相关文章
|
存储 PyTorch 算法框架/工具
Tensor to img && imge to tensor (pytorch的tensor转换)
Tensor to img && imge to tensor (pytorch的tensor转换)
|
2月前
|
机器学习/深度学习 Python
tf.keras.layers.LSTM/tf.keras.layers.LSTMCell
【8月更文挑战第20天】tf.keras.layers.LSTM/tf.keras.layers.LSTMCell。
34 1
|
2月前
tf.random
【8月更文挑战第12天】tf.random。
31 3
|
11天前
|
TensorFlow 算法框架/工具
Tensorflow error(二):x and y must have the same dtype, got tf.float32 != tf.int32
本文讨论了TensorFlow中的一个常见错误,即在计算过程中,变量的数据类型(dtype)不一致导致的错误,并通过使用`tf.cast`函数来解决这个问题。
13 0
|
2月前
|
算法框架/工具 数据格式
tf.keras.layers.Conv2D
【8月更文挑战第20天】tf.keras.layers.Conv2D。
31 2
|
2月前
tf.keras.layers.Dense
【8月更文挑战第20天】tf.keras.layers.Dense。
44 2
|
2月前
tf.keras.layers.MaxPooling2D/AveragePooling2D
【8月更文挑战第19天】tf.keras.layers.MaxPooling2D/AveragePooling2D。
30 0
|
机器学习/深度学习 PyTorch API
Torch
Torch是一个用于构建深度学习模型的开源机器学习库,它基于Lua编程语言。然而,由于PyTorch的出现,现在通常所说的"torch"指的是PyTorch。PyTorch是一个基于Torch的Python库,它提供了一个灵活而高效的深度学习框架。
258 1
怎么将[tensor([[ 1, 2]]), tensor([[5, 6]]), tensor([[9, 10]])] 合并成 tensor([[1,2],[3,4],[5,6]])
可以先使用 torch.cat() 函数将列表中的张量在第0维(行)上进行拼接,然后再使用 .view() 函数将形状调整为需要的形状。
184 0
torch中如何将tensor([[1, 2, 3]]) 和 tensor([4]) 合并成 tensor([[1,2,3,4]])
可以使用 torch.cat() 方法将两个张量沿着指定的维度进行拼接
467 0