mindspore.ops.Gather
- class mindspore.ops.Gather(batch_dims=0)[源代码]
prim = ops.Gather(batch_dims) out = prim(input_params, input_indices, axis)
等价于
ops.gather(input_params, input_indices, axis, batch_dims)
更多详情请查看:
mindspore.ops.gather()
。
prim = ops.Gather(batch_dims)
out = prim(input_params, input_indices, axis)
等价于
ops.gather(input_params, input_indices, axis, batch_dims)
更多详情请查看: mindspore.ops.gather()
。