TensorFlow与MindSpore

查看源文件

由社区提供的TensorFlow APIs和MindSpore APIs之间的映射,可能在参数、输入、输出、逻辑功能和特定场景等方面存在差异,可详见各API描述或已提供的差异对比。

也欢迎更多的MindSpore开发者参与完善映射内容。

tf

tf.distribute

tf.image

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.image.ssim

mindspore.nn.SSIM

tf.keras

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.keras.initializers.Constant

mindspore.common.initializer.Constant

差异对比

tf.keras.initializers.Ones

mindspore.common.initializer.One

tf.keras.initializers.RandomNormal

mindspore.common.initializer.Normal

差异对比

tf.keras.initializers.TruncatedNormal

mindspore.common.initializer.TruncatedNormal

差异对比

tf.keras.initializers.RandomUniform

mindspore.common.initializer.Uniform

差异对比

tf.keras.initializers.VarianceScaling

mindspore.common.initializer.XavierUniform

差异对比

tf.keras.initializers.Zeros

mindspore.common.initializer.Zero

tf.keras.backend.batch_dot

mindspore.ops.batch_dot

tf.keras.backend.dot

mindspore.ops.dot

tf.keras.layers.Embedding

mindspore.nn.Embedding

tf.keras.layers.Flatten

mindspore.nn.Flatten

tf.keras.layers.LayerNormalization

mindspore.nn.LayerNorm

tf.keras.layers.LSTM

mindspore.nn.LSTM

tf.keras.layers.PReLU

mindspore.nn.PReLU

tf.keras.layers.RNN

mindspore.ops.DynamicRNN

tf.keras.Model

mindspore.Model

差异对比

tf.keras.Model.fit
tf.keras.Model.fit_generator

mindspore.Model.train

差异对比

tf.keras.Model.predict
tf.keras.Model.predict_generator

mindspore.Model.eval

差异对比

tf.keras.optimizers.Adagrad

mindspore.nn.Adagrad

差异对比

tf.keras.optimizers.Adam

mindspore.nn.Adam

差异对比

tf.keras.optimizers.Ftrl

mindspore.nn.FTRL

差异对比

tf.keras.optimizers.RMSprop

mindspore.nn.RMSProp

差异对比

tf.keras.optimizers.SGD

mindspore.nn.SGD

差异对比

tf.keras.metrics.Accuracy
tf.keras.metrics.BinaryAccuracy
tf.keras.metrics.CategoricalAccuracy
tf.keras.metrics.SparseCategoricalAccuracy

mindspore.nn.Accuracy

差异对比

tf.keras.metrics.AUC

mindspore.nn.auc

差异对比

tf.keras.metrics.CosineSimilarity

mindspore.nn.CosineSimilarity

差异对比

tf.keras.metrics.Mean

mindspore.nn.Loss

tf.keras.metrics.MeanAbsoluteError

mindspore.nn.MAE

tf.keras.metrics.MeanSquaredError

mindspore.nn.MSE

tf.keras.metrics.Precision

mindspore.nn.Precision

差异对比

tf.keras.metrics.Recall

mindspore.nn.Recall

差异对比

tf.layers

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.layers.Dense

mindspore.nn.Dense

tf.linalg

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.linalg.inv

mindspore.ops.Inv

tf.linalg.matmul

mindspore.nn.MatMul

tf.math

tf.metrics

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.metrics.mean_iou

mindspore.ops.IOU

tf.nn

tf.random

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.random.gamma

mindspore.ops.Gamma

tf.random.uniform

mindspore.ops.uniform

tf.sparse

TensorFlow 1.15 APIs

MindSpore APIs

说明

tf.sparse.SparseTensor

mindspore.SparseTensor

tf.train

tfp

TensorFlow Probability 0.14.1 APIs

MindSpore APIs

说明

tfp.bijectors.Softplus

mindspore.nn.probability.bijector.Softplus

差异对比