嗨,我是tensorflow的新手.我想在tensorflow中实现以下python代码.

import numpy as np
a = np.array([1,2,3,4,5,6,7,9,0])
print(a) ## [1 2 3 4 5 6 7 9 0]
print(a.shape) ## (9,)
b = a[:, np.newaxis] ### want to write this in tensorflow.
print(b.shape) ## (9,1)

推荐答案

我想应该是tf.expand_dims美元-

tf.expand_dims(a, 1) # Or tf.expand_dims(a, -1)

基本上,我们列出了要插入新轴的轴ID,后面的轴/DIM为pushed-back.

从链接的文档中,这里有几个扩展维度的例子-

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]

Python-3.x相关问答推荐

删除括号和大括号中不必要的空格

BeautifulSoup 和 pd.read_html - 如何将链接保存到最终数据框中的单独列中?

列表中的重复数字与列表理解

如何将数据框中的每一行转换为具有属性的节点?

在“with”语句中调用构造函数

“警告:scheme.data 的值不匹配”当我try 更新 pip 或安装软件包时

逗号分隔列表的 argparse 操作或类型

如何使用存储在变量中的值作为 case 模式?

在 Python 中判断布尔环境变量

没有找到 Python;不带参数运行以从 Microsoft Store 安装,或从“设置”中禁用此快捷方式

如何用一个 for 循环和没有 if 条件来对偶数和奇数求和?

如何生成满足某些条件的三个随机整数?

Python 3.x 中的 NoneType 在哪里?

NumPy ndarray dtype 的类型提示?

理解 Keras 的 ImageDataGenerator 类中的 `width_shift_range` 和 `height_shift_range` 参数

如何用decorator 绕过python函数定义?

*(单星)和/(斜线)作为独立参数有什么作用?

为什么 f'{{{74}}}' 与 f'{{74}}' 与 f-Strings 相同?

如何解决“No module named 'frontend'”错误消息?

Plotly:如何设置线条 colored颜色 ?