【Mindspore模型迁移指北】

社会演员多

2 年前

【笔者按】 笔者最近在进行讲Mask_RCNN网络迁移到Mindspore框架的工作。本文会按照个人进度迭代更新，记录所学所思。

Table of Contents

1. Mindspore算子

mindspore.nn
神经网络Cell。用于构建神经网络中的预定义构建块或计算单元。

mindspore.ops
可用于Cell的构造函数的算子。

import mindspore.ops as ops

标量与Tensor运算：运算时，将标量自动转为Tensor，数据类型和参与运算的Tensor数据类型保持一致；当Tensor是bool数据类型，标量是int或float时，将标量和Tensor都转为数据类型为int32或float32的Tensor；当Tensor是int或者uint数据类型，标量是float时，将标量和Tensor都转为数据类型为float32的Tensor。
不同数据类型Tensor运算：数据类型优先级排序为bool < uint8 < int8 < int16 < int32 < int64 < float16 < float32 <float64，运算时，先确定参与运算的Tensor中优先级相对最高的数据类型，然后将低优先级数据类型Tensor转换为相对最高优先级数据类型；而当int8和uint8数据类型的Tensor进行运算时，将其都转为int16的Tensor。
不支持对Parameter进行数据类型转换：如果按照转换规则推导，需要对网络中定义的Parameter进行数据类型转换时，会抛出RuntimeError异常。

import tensorflow as tf
import mindspore as ms

tf.cast(scalar,tensorflow.dtype)<--->ms.ops.ScalarCast(scalar,mindspore.dtype)

文章出处登录后可见！

已经登录？立即刷新