mxnet,nd中的asscalar()函数

X.asscalar()

将向量X转换成标量,且向量X只能为一维含单个元素的向量

X:nd类型的数据

创建一个单个元素的向量

test=nd.array([2])

输出:
[2.]
<NDArray 1 @cpu(0)>

转换成标量:

test.asscalar()

输出:2.0

版权声明:本文为shenkunchang1877原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。