list需要转换成numpy数组获得维度
但是,可能会遇到报错 “Cannot convert list to array: ValueError: only one element tensors can be converted to Python scalar”:
原因是想将多个网络输出的结果(tensor类型)放到一个python list中, 然后直接转换成numpy类型, 这样会报错, 问题在于只能将一个含有一个元素的Tensor转换成python标量。
解决方法:对每一tensor先转换成numpy类型,然后在进行操作
例1
例2