登录 注册
开源 企业版 高校版 私有云 Gitee AI NEW
开源项目 > 其他开源 > 图书/手册/教程 && 人工智能 > 机器学习/深度学习 &&
扫描微信二维码支付
取消
支付完成
Watch
不关注 关注所有动态 仅关注版本发行动态 关注但不提醒动态
335 Star 1.5K Fork 861

MindSpore / docs

代码 Issues 64 Pull Requests 5 Wiki 统计 流水线
服务
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目，私有仓库也完全免费 ：）
免费加入
克隆/下载
tensorflow_api_mapping.md 56.38 KB
一键复制 编辑 原始数据 按行查看 历史
宦晓玲 提交于 2023-07-21 16:52 . modify the md links 1.8

TensorFlow and MindSpore

View Source On Gitee

Mapping between TensorFlow APIs and MindSpore APIs, which is provided by the community. There may be differences in parameters, inputs, outputs, logic functions, and specific scenarios. For details, see the description of each API or the difference comparison provided.

More MindSpore developers are also welcome to participate in improving the mapping content.

tf

TensorFlow 1.15 APIs MindSpore APIs Description
tf.expand_dims mindspore.ops.ExpandDims
tf.eye mindspore.ops.Eye
tf.fill mindspore.ops.Fill
tf.gather mindspore.ops.Gather
tf.gradients mindspore.ops.GradOperation diff
tf.norm mindspore.nn.Norm
tf.one_hot mindspore.nn.OneHot
tf.ones_like mindspore.ops.OnesLike
tf.pad mindspore.nn.Pad
tf.print mindspore.ops.Print
tf.reshape mindspore.ops.Reshape
tf.shape mindspore.ops.Shape
tf.size mindspore.ops.Size
tf.slice mindspore.ops.Slice
tf.stop_gradient mindspore.ops.stop_gradient
tf.Tensor mindspore.Tensor
tf.tile mindspore.ops.Tile
tf.transpose mindspore.ops.Transpose
tf.zeros_like mindspore.ops.ZerosLike

tf.data

Many data processing operations in MindSpore are shared by each dataset class. The following uses ImageFolderDataset and CSVDataset as examples to compare the API differences, and other dataset classes are the same.

TensorFlow 1.15 APIs MindSpore APIs Description
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 diff
tf.data.Dataset.from_tensor_slices mindspore.dataset.NumpySlicesDataset diff
tf.data.Dataset.map mindspore.dataset.GeneratorDataset.map
tf.data.Dataset.prefetch mindspore.dataset.config.set_prefetch_size diff
tf.data.Dataset.repeat mindspore.dataset.GeneratorDataset.repeat
tf.data.Dataset.shuffle mindspore.dataset.GeneratorDataset.shuffle diff
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 diff
tf.data.TFRecordDataset mindspore.dataset.TFRecordDataset diff
tf.data.experimental.bucket_by_sequence_length mindspore.dataset.GeneratorDataset.bucket_batch_by_length diff
tf.data.experimental.CsvDataset mindspore.dataset.CSVDataset diff

tf.distribute

TensorFlow 1.15 APIs MindSpore APIs Description
tf.distribute.Strategy set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL) diff

tf.image

TensorFlow 1.15 APIs MindSpore APIs Description
tf.image.ssim mindspore.nn.SSIM
tf.image.central_crop mindspore.dataset.vision.CenterCrop diff
tf.image.convert_image_dtype mindspore.dataset.transforms.TypeCast diff
tf.image.crop_to_bounding_box mindspore.dataset.vision.Crop diff
tf.image.flip_left_right mindspore.dataset.vision.HorizontalFlip
tf.image.flip_up_down mindspore.dataset.vision.VerticalFlip
tf.image.grayscale_to_rgb mindspore.dataset.vision.ConvertColor diff
tf.image.hsv_to_rgb mindspore.dataset.vision.HsvToRgb
tf.image.pad_to_bounding_box mindspore.dataset.vision.Pad diff
tf.image.per_image_standardization mindspore.dataset.vision.Normalize diff
tf.image.random_crop mindspore.dataset.vision.RandomCrop diff
tf.image.random_flip_left_right mindspore.dataset.vision.RandomHorizontalFlip diff
tf.image.random_flip_up_down mindspore.dataset.vision.RandomVerticalFlip diff
tf.image.ResizeMethod mindspore.dataset.vision.Inter
tf.image.resize_images mindspore.dataset.vision.Resize diff
tf.image.rgb_to_grayscale mindspore.dataset.vision.ConvertColor diff
tf.image.rot90 mindspore.dataset.vision.Rotate diff

tf.io

TensorFlow 1.15 APIs MindSpore APIs Description
tf.io.decode_image mindspore.dataset.vision.Decode diff

tf.keras

TensorFlow 1.15 APIs MindSpore APIs Description
tf.keras.backend.batch_dot mindspore.ops.batch_dot
tf.keras.backend.dot mindspore.ops.dot
tf.keras.datasets.cifar10 mindspore.dataset.Cifar10Dataset diff
tf.keras.datasets.cifar100 mindspore.dataset.Cifar100Dataset diff
tf.keras.datasets.fashion_mnist mindspore.dataset.FashionMnistDataset diff
tf.keras.datasets.imdb mindspore.dataset.IMDBDataset diff
tf.keras.datasets.mnist mindspore.dataset.MnistDataset diff
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.optimizers.Adagrad mindspore.nn.Adagrad diff
tf.keras.optimizers.Adam mindspore.nn.Adam diff
tf.keras.optimizers.SGD mindspore.nn.SGD diff
tf.keras.preprocessing.image.random_rotation mindspore.dataset.vision.RandomRotation diff
tf.keras.preprocessing.image.random_shear mindspore.dataset.vision.RandomAffine diff
tf.keras.preprocessing.image.random_shift mindspore.dataset.vision.RandomAffine diff

tf.layers

TensorFlow 1.15 APIs MindSpore APIs Description
tf.layers.Dense mindspore.nn.Dense

tf.linalg

TensorFlow 1.15 APIs MindSpore APIs Description
tf.linalg.matmul mindspore.nn.MatMul

tf.math

TensorFlow 1.15 APIs MindSpore APIs Description
tf.math.add mindspore.ops.Add
tf.math.add_n mindspore.ops.AddN
tf.math.divide mindspore.ops.Div
tf.math.erf mindspore.ops.Erf
tf.math.greater mindspore.ops.Greater
tf.math.less_equal mindspore.ops.LessEqual
tf.math.log mindspore.ops.Log
tf.math.multiply mindspore.ops.Mul
tf.math.pow mindspore.ops.Pow
tf.math.sigmoid mindspore.nn.Sigmoid
tf.math.subtract mindspore.ops.Sub

tf.metrics

TensorFlow 1.15 APIs MindSpore APIs Description
tf.metrics.mean_iou mindspore.ops.IOU

tf.nn

TensorFlow 1.15 APIs MindSpore APIs Description
tf.nn.avg_pool2d mindspore.nn.AvgPool2d
tf.nn.batch_normalization mindspore.ops.BatchNorm
tf.nn.bias_add mindspore.ops.BiasAdd
tf.nn.conv2d mindspore.nn.Conv2d
tf.nn.conv2d_transpose mindspore.nn.Conv2dTranspose
tf.nn.ctc_loss mindspore.ops.CTCLoss
tf.nn.dropout mindspore.nn.Dropout
tf.nn.elu mindspore.nn.ELU
tf.nn.l2_loss mindspore.ops.L2Loss
tf.nn.l2_normalize mindspore.ops.L2Normalize
tf.nn.leaky_relu mindspore.nn.LeakyReLU
tf.nn.max_pool_with_argmax mindspore.ops.MaxPoolWithArgmax
tf.nn.max_pool2d mindspore.nn.MaxPool2d
tf.nn.moments mindspore.nn.Moments
tf.nn.relu mindspore.nn.ReLU
tf.nn.selu mindspore.ops.SeLU
tf.nn.sigmoid_cross_entropy_with_logits mindspore.ops.SigmoidCrossEntropyWithLogits
tf.nn.softmax mindspore.nn.Softmax
tf.nn.softmax_cross_entropy_with_logits mindspore.nn.SoftmaxCrossEntropyWithLogits diff

tf.random

TensorFlow 1.15 APIs MindSpore APIs Description
tf.random.gamma mindspore.ops.Gamma
tf.random.uniform mindspore.ops.uniform

tf.sparse

TensorFlow 1.15 APIs MindSpore APIs Description
tf.sparse.SparseTensor mindspore.SparseTensor

tf.train

TensorFlow 1.15 APIs MindSpore APIs Description
tf.train.exponential_decay mindspore.nn.ExponentialDecayLR
tf.train.linear_cosine_decay mindspore.nn.CosineDecayLR diff
tf.train.MomentumOptimizer mindspore.nn.Momentum diff

tfp

TensorFlow Probability 0.14.1 APIs MindSpore APIs Description
tfp.bijectors.Softplus mindspore.nn.probability.bijector.Softplus diff
1
https://gitee.com/mindspore/docs.git
git@gitee.com:mindspore/docs.git
mindspore
docs
docs
r1.8
点此查找更多帮助

搜索帮助

Git 命令在线学习 如何在 Gitee 导入 GitHub 仓库
Git 仓库基础操作
企业版和社区版功能对比
SSH 公钥设置
如何处理代码冲突
仓库体积过大，如何减小？
如何找回被删除的仓库数据
Gitee 产品配额说明
GitHub仓库快速导入Gitee及同步更新
什么是 Release（发行版）
将 PHP 项目自动发布到 packagist.org
仓库举报
回到顶部