コード例 #1
0
 /// <inheritdoc/>
 Tensor IOps.ScatterND(Tensor X, Tensor indices, Tensor updates, Layer.ScatterNDReductionMode reduction)
 {
     var O = m_Ops.ScatterND(X, indices, updates, reduction);
     LogLayerSummary("{" + X.shape + "," + indices.shape + "," + updates.shape + "," + reduction + "} # " + O.shape);
     LogOutputTensorSummary(O, Prefix + "Gather");
     return O;
 }
コード例 #2
0
        // <inheritdoc/>
        Tensor IOps.ScatterND(Tensor X, Tensor indices, Tensor updates, Layer.ScatterNDReductionMode reduction)
        {
            var Y = m_Ops1.ScatterND(X, indices, updates, reduction);
            var Z = m_Ops2.ScatterND(X, indices, updates, reduction);

            CheckSame(Y, Z, Layer.Type.ScatterND);
            return(Y);
        }
コード例 #3
0
ファイル: StatsOps.cs プロジェクト: mantasp/barracuda-release
        // <inheritdoc/>
        Tensor IOps.ScatterND(Tensor X, Tensor indices, Tensor updates, Layer.ScatterNDReductionMode reduction)
        {
            var O = m_Ops.ScatterND(X, indices, updates, reduction);

            Elementwise(O);
            RegisterLayerStats();
            return(O);
        }