TensorFlow与MindSpore

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

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

tf.data

MindSpore中的许多数据处理方法为各数据集类共有,下面以 GeneratorDataset 为例进行API差异对比,其他数据集类也是同理。

TensorFlow 1.15 APIs MindSpore APIs 说明
tf.data.Dataset.apply mindspore.dataset.GeneratorDataset.apply
tf.data.Dataset.batch mindspore.dataset.GeneratorDataset.batch
tf.data.Dataset.concatenate mindspore.dataset.GeneratorDataset.concat
tf.data.Dataset.filter mindspore.dataset.GeneratorDataset.filter
tf.data.Dataset.flat_map mindspore.dataset.GeneratorDataset.flat_map
tf.data.Dataset.from_generator mindspore.dataset.GeneratorDataset 差异对比
tf.data.Dataset.from_tensor_slices mindspore.dataset.NumpySlicesDataset 差异对比
tf.data.Dataset.map mindspore.dataset.GeneratorDataset.map
tf.data.Dataset.prefetch mindspore.dataset.config.set_prefetch_size 差异对比
tf.data.Dataset.repeat mindspore.dataset.GeneratorDataset.repeat
tf.data.Dataset.shuffle mindspore.dataset.GeneratorDataset.shuffle 差异对比
tf.data.Dataset.skip mindspore.dataset.GeneratorDataset.skip
tf.data.Dataset.take mindspore.dataset.GeneratorDataset.take
tf.data.Dataset.zip mindspore.dataset.GeneratorDataset.zip
tf.data.TextLineDataset mindspore.dataset.TextFileDataset 差异对比
tf.data.TFRecordDataset mindspore.dataset.TFRecordDataset 差异对比
tf.data.experimental.bucket_by_sequence_length mindspore.dataset.GeneratorDataset.bucket_batch_by_length 差异对比
tf.data.experimental.CsvDataset mindspore.dataset.CSVDataset 差异对比

tf.io

TensorFlow 1.15 APIs MindSpore APIs 说明
tf.io.decode_image mindspore.dataset.vision.Decode 差异对比

tf.keras

TensorFlow 1.15 APIs MindSpore APIs 说明
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.backend.batch_dot mindspore.ops.batch_dot
tf.keras.backend.dot mindspore.ops.dot
tf.keras.datasets.cifar10 mindspore.dataset.Cifar10Dataset 差异对比
tf.keras.datasets.cifar100 mindspore.dataset.Cifar100Dataset 差异对比
tf.keras.datasets.fashion_mnist mindspore.dataset.FashionMnistDataset 差异对比
tf.keras.datasets.imdb mindspore.dataset.IMDBDataset 差异对比
tf.keras.datasets.mnist mindspore.dataset.MnistDataset 差异对比
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.RandomUniform mindspore.common.initializer.Uniform 差异对比
tf.keras.initializers.TruncatedNormal mindspore.common.initializer.TruncatedNormal 差异对比
tf.keras.initializers.VarianceScaling mindspore.common.initializer.XavierUniform 差异对比
tf.keras.initializers.Zeros mindspore.common.initializer.Zero
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.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.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.preprocessing.image.random_rotation mindspore.dataset.vision.RandomRotation 差异对比
tf.keras.preprocessing.image.random_shear mindspore.dataset.vision.RandomAffine 差异对比
tf.keras.preprocessing.image.random_shift mindspore.dataset.vision.RandomAffine 差异对比

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.metrics

TensorFlow 1.15 APIs MindSpore APIs 说明
tf.metrics.mean_iou mindspore.ops.IOU

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

tfp

TensorFlow Probability 0.14.1 APIs MindSpore APIs 说明
tfp.bijectors.Softplus mindspore.nn.probability.bijector.Softplus 差异对比