mindspore.dataset.transforms.Concatenate

查看源文件
class mindspore.dataset.transforms.Concatenate(axis=0, prepend=None, append=None)[源代码]

在输入数据的某一个轴上进行数组拼接,目前仅支持拼接形状为1D的数组。

参数:
  • axis (int, 可选) - 指定一个轴用于拼接数组。默认值: 0

  • prepend (numpy.ndarray, 可选) - 想要拼接到输入数据首部的数组。默认值: None ,不在首部拼接数组。

  • append (numpy.ndarray, 可选) - 想要拼接到输入数据尾部的数组。默认值: None ,不在尾部拼接数组。

异常:
  • TypeError - 参数 axis 的类型不为int。

  • TypeError - 参数 prepend 的类型不为 numpy.ndarray

  • TypeError - 参数 append 的类型不为 numpy.ndarray

支持平台:

CPU

样例:

>>> import numpy as np
>>> import mindspore.dataset as ds
>>> import mindspore.dataset.transforms as transforms
>>>
>>> # Use the transform in dataset pipeline mode
>>> # concatenate string
>>> prepend_tensor = np.array(["dw", "df"])
>>> append_tensor = np.array(["dwsdf", "df"])
>>> concatenate_op = transforms.Concatenate(0, prepend_tensor, append_tensor)
>>> data = [["This","is","a","string"]]
>>> numpy_slices_dataset = ds.NumpySlicesDataset(data)
>>> numpy_slices_dataset = numpy_slices_dataset.map(operations=concatenate_op)
>>> for item in numpy_slices_dataset.create_dict_iterator(num_epochs=1, output_numpy=True):
...     print(item["column_0"].shape, item["column_0"].dtype)
(8,) <U6
>>>
>>> # Use the transform in eager mode
>>> data = np.array([1, 2, 3])
>>> prepend_tensor = np.array([10, 20])
>>> append_tensor = np.array([100])
>>> output = transforms.Concatenate(0, prepend_tensor, append_tensor)(data)
>>> print(output.shape, output.dtype)
(6,) int64