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

小知识点,当作学习笔记记录一下。


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)
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加了一个行的维度。

你可能感兴趣的:(Python,tensorflow,莫烦视频学习,numpy,newaxis,linspace)