---
title: tf.train.batch接口手工迁移
description: "TensorFlow不建议使用tf.train.batch接口通过队列形式处理输入数据，建议替换为tf.data.Dataset.batch(batch_size)。"
url: https://www.hiascend.com/document/detail/zh/TensorFlowCommercial/latest/migration/tfmigr1/tfmigr1_000115.html
sourcePath: /source/zh/TensorFlowCommercial/900/migration/tfmigr1/tfmigr1_000115.html
indexId: 0374f287f784022d65ba7dbc7c021179b058455285caa64ad88653d805aee87c73
---
# tf.train.batch接口手工迁移

#### 迁移原因

TensorFlow不建议使用tf.train.batch接口通过队列形式处理输入数据，建议替换为tf.data.Dataset.batch(batch_size)。

如果用户仍然希望在CANN上使用tf.train.batch接口，那么建议num_threads参数配置为1。


#### 迁移示例

原始脚本：

```
(texts, texts_tests, mels, mags, dones) = tf.train.batch([text, texts_test, mel, mag, done], shapes=[(hp.T_x,), (hp.T_x,), ((hp.T_y // hp.r), (hp.n_mels * hp.r)), (hp.T_y, (1 + (hp.n_fft // 2))), ((hp.T_y // hp.r),)], num_threads=2, batch_size=batch_size, capacity=(batch_size * 8), dynamic_pad=False)
```

迁移后脚本：

```
(texts, texts_tests, mels, mags, dones) = tf.train.batch([text, texts_test, mel, mag, done], shapes=[(hp.T_x,), (hp.T_x,), ((hp.T_y // hp.r), (hp.n_mels * hp.r)), (hp.T_y, (1 + (hp.n_fft // 2))), ((hp.T_y // hp.r),)], num_threads=1, batch_size=batch_size, capacity=(batch_size * 8), dynamic_pad=False)
```