array的axis

对于array,从最外面的[ ]是0轴axis = 0,一层一层往里是1,2,3。。轴
reshape,transpose这些函数的参数都是针对axis来说的
比如说transpose((2,2,3))
这是说0轴2个,1轴2个,2轴3个
[ [ [1,2,3], [1,2,3] ], [ [1,2,3], [1,2,3] ] ]

这样理解准没错,包括Tensor(5,3)这样的也是,第一个[ ]中是5个元素,向里在一个 [ ]中是3个元素

猜你喜欢

转载自blog.csdn.net/Innovat1on/article/details/88845203