维度 Tensor([]) Tensor([ [ [] ] ])"/>
维度 Tensor([]) Tensor([ [ [] ] ])
Tensor(0) : 代表的是一个普通的数字0,维度是0.
Tensor( [0] ) : 代表是一个向量,维度是1。有几组中括号,维度就是几。只不过如果里面只有一个数字的话,这个时候恰巧它只有一个数字而已,不像矩阵一样的,有很多数字。
总结:一个Tensor的维度是多少,它的size或者shape的向量的长度就是多少。假如一个Tensor的维度是3,那肯定它的形状要把3个维度上的形状都告诉你。一般,一个数据点的数据是一个向量,是一维的。全连接网络的数据是二维的,是一个向量,是行数×特征数量。CNN的输入数据是四维的,是batch size × channel数量 × 宽 × 高。
更多推荐
维度 Tensor([]) Tensor([ [ [] ] ])
发布评论