mindspore.dataset.transforms.Concatenate

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

在Tensor的某一个轴上进行元素拼接,目前仅支持拼接形状为1D的Tensor。

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

  • prepend (numpy.ndarray, 可选) - 指定拼接在最前面的Tensor。默认值: None ,不指定。

  • append (numpy.ndarray, 可选) - 指定拼接在最后面的Tensor。默认值: None ,不指定。

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

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

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

支持平台:

CPU

样例:

>>> import mindspore.dataset as ds
>>> import mindspore.dataset.transforms as transforms
>>> import numpy as np
>>>
>>> # concatenate string
>>> prepend_tensor = np.array(["dw", "df"], dtype='S')
>>> append_tensor = np.array(["dwsdf", "df"], dtype='S')
>>> concatenate_op = transforms.Concatenate(0, prepend_tensor, append_tensor)
>>> data = [["This","is","a","string"]]
>>> dataset = ds.NumpySlicesDataset(data)
>>> dataset = dataset.map(operations=concatenate_op)