一维张量的基本属性 import numpy as np a = np . array ( [ 1 , 5 , 3 , 9 , 2 ] ) print ( a . mean ( ) ) #求平均值 print ( a . max ( ) ) print ( a . min ( ) ) print ( a . argmax ( ) ) #输出最大元素的下标,从0开始 print ( a * 2 ) print ( a . ndim ) 二维张量的基本属性 import numpy as np a = np . array ( [ [ 1 , 2 , 3 ] , [ 7 , 3 , 4 ] , [ 5 , 0 , 1 ] ] ) #几维张量就有几个中括号 print ( a . shape ) print ( a . argmax ( ) ) #输出的结果是一个平铺后的下标,从0开始 print ( a . flatten ( ) ) #平铺为一行 print ( a . reshape ( 9 , 1 ) ) 来源: CSDN 作者: skyline_ss 链接: https://blog.csdn.net/skylinesssss/article/details/103460677