edd86f6318b411a42a1f287fa6359bbfd12fa71c,python/src/nnabla/functions.py,,scatter_nd,#Any#Any#Any#Any#,915

Before Change

    if ref is not None:
        if not isinstance(ref, nn.NdArray):
            raise TypeError("reference data must be NdArray type")
        if not ref.shape == shape:
            raise ValueError("reference data must have shape {}".format(shape))
    return scatter_nd_base(data, indices, shape, outputs=[ref])

After Change

        indices = nn.Variable.from_numpy_array(indices)
    if shape is None and out is None:
        raise TypeError("One of `shape` or `out` argument must be supplied.")
    if shape and out:
        raise TypeError("Only one of `shape` or `out` argument may be used.")
    if out:
        if isinstance(out, nn.Variable):
            out = out.data
        if not isinstance(out, nn.NdArray):
            raise TypeError("`out` argument must be NdArray or Variable type.")
        shape = out.shape
        outputs = [out]
    else:
        if isinstance(shape, np.ndarray):
            shape = shape.tolist()
        outputs = None
    return scatter_nd_base(data, indices, shape, outputs=outputs)

Instances