您的位置:首页 > 编程语言 > Python开发

tensorflow学习心得:numpy中newaxis的使用

2018-01-15 17:46 134 查看
小知识点,当作学习笔记记录一下。

newaxis的功能其实很简单,主要就是加入一个新的维度,大家可以参照我下面给出的例子,自行理解!

示例:

import numpy as np

x_data = np.linspace(-1,1,3)
y_data = np.linspace(-1,1,3)[:,np.newaxis]
z_data = np.linspace(-1,1,3)[np.newaxis,:]
print(x_data.shape)
print(y_data.shape)
print(z_data.shape)
print(x_data)
aa17
print(y_data)
print(z_data)


结果:

(3,)
(3, 1)
(1, 3)
[-1.  0.  1.]
[[-1.]
[ 0.]
[ 1.]]
[[-1.  0.  1.]]


注:linspace函数的功能是创建等差数列,例子中给定了三个参数,意思分别是开始元素,结束元素,生成等差数列中的元素个数。我给的例子中的意思就是从-1到1,生成具有3个元素的等差数列,所以生成的结果为[-1. 0. 1.]。

y_data为x_data在列中加了一个列的维度;z_data为x_data加了一个行的维度。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: