TensorFlow与MindSpore API映射表

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

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

TensorFlow 2.6

tf

TensorFlow 2.6 APIs MindSpore APIs 说明
tf.bitwise.invert mindspore.ops.invert 一致
tf.clip_by_value mindspore.ops.clip_by_value 差异对比
tf.expand_dims mindspore.ops.expand_dims 差异对比
tf.eye mindspore.ops.eye 差异对比
tf.fill mindspore.ops.fill 差异对比
tf.range mindspore.ops.range 功能一致,参数名不同
tf.random.uniform_candidate_sampler mindspore.ops.uniform_candidate_sampler 差异对比
tf.raw_ops.ApplyGradientDescent mindspore.ops.ApplyGradientDescent 一致
tf.raw_ops.ApproximateEqual mindspore.ops.approximate_equal 一致
tf.raw_ops.DataFormatDimMap mindspore.ops.DataFormatDimMap 功能一致,参数名不同
tf.raw_ops.InplaceUpdate mindspore.ops.inplace_update 功能一致,参数名不同
tf.raw_ops.Inv mindspore.ops.inv 一致
tf.raw_ops.LRN mindspore.ops.LRN 差异对比
tf.raw_ops.MatrixDiagPartV3 mindspore.ops.matrix_diag_part 功能一致,参数名不同
tf.raw_ops.MatrixDiagV3 mindspore.ops.matrix_diag 功能一致,参数名不同
tf.raw_ops.MatrixSetDiagV3 mindspore.ops.matrix_set_diag 功能一致,参数名不同
tf.raw_ops.MatrixSolve mindspore.ops.matrix_solve 一致
tf.raw_ops.RandomGamma mindspore.ops.random_gamma 一致
tf.raw_ops.ScatterNdMax mindspore.ops.scatter_nd_max 功能一致,参数名不同
tf.raw_ops.ScatterNdMin mindspore.ops.scatter_nd_min 功能一致,参数名不同
tf.raw_ops.Select mindspore.ops.select 功能一致,参数名不同
tf.raw_ops.SparseApplyAdagradV2 mindspore.ops.SparseApplyAdagradV2 差异对比
tf.raw_ops.SparseToDense mindspore.ops.SparseToDense 差异对比
tf.raw_ops.TensorScatterAdd mindspore.ops.tensor_scatter_add 功能一致,参数名不同
tf.raw_ops.TensorScatterSub mindspore.ops.tensor_scatter_sub 功能一致,参数名不同
tf.reverse_sequence mindspore.ops.reverse_sequence 功能一致,参数名不同
tf.scatter_nd mindspore.ops.scatter_nd 一致
tf.space_to_batch_nd mindspore.ops.space_to_batch_nd 功能一致,参数名不同
tf.unstack mindspore.ops.unstack 功能一致,参数名不同

tf.data

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

TensorFlow 2.6 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 差异对比

TensorFlow 1.15

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.train.Model 差异对比
tf.keras.Model.fit
tf.keras.Model.fit_generator
mindspore.train.Model.train 差异对比
tf.keras.Model.predict
tf.keras.Model.predict_generator
mindspore.train.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.RNN mindspore.ops.DynamicRNN
tf.keras.metrics.Accuracy
tf.keras.metrics.BinaryAccuracy
tf.keras.metrics.CategoricalAccuracy
tf.keras.metrics.SparseCategoricalAccuracy
mindspore.train.Accuracy 差异对比
tf.keras.metrics.AUC mindspore.train.auc 差异对比
tf.keras.metrics.CosineSimilarity mindspore.train.CosineSimilarity 差异对比
tf.keras.metrics.Mean mindspore.train.Loss
tf.keras.metrics.MeanAbsoluteError mindspore.train.MAE
tf.keras.metrics.MeanSquaredError mindspore.train.MSE
tf.keras.metrics.Precision mindspore.train.Precision 差异对比
tf.keras.metrics.Recall mindspore.train.Recall 差异对比
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.linalg

TensorFlow 1.15 APIs MindSpore APIs 说明
tf.linalg.matmul mindspore.ops.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 差异对比