mindspore.Tensor.init_data
- mindspore.Tensor.init_data(slice_index=None, shape=None, opt_shard_group=None)[源代码]
获取此Tensor的数据。
说明
对于同一个Tensor,只可以调用一次 init_data 函数。
- 参数:
slice_index (int) - 参数切片的索引。在初始化参数切片的时候使用,保证使用相同切片的设备可以生成相同的Tensor。默认值:
None
。shape (list[int]) - 切片的shape,在初始化参数切片时使用。默认值:
None
。opt_shard_group (str) - 优化器分片组,在自动或半自动并行模式下用于获取参数的切片。关于优化器分组,请参考 优化器并行 。默认值:
None
。
- 返回:
初始化的Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore as ms >>> from mindspore.common.initializer import initializer, Constant >>> x = initializer(Constant(1), [2, 2], ms.float32) >>> out = x.init_data() >>> print(out) [[1. 1.] [1. 1.]]