org.nd4j.linalg.api.ops.BaseBroadcastOp Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation

There is a newer version: 1.0.0-M2.1

package org.nd4j.linalg.api.ops;

import lombok.NoArgsConstructor;
import org.nd4j.linalg.api.ndarray.INDArray;

@NoArgsConstructor
public abstract class BaseBroadcastOp extends BaseOp implements BroadcastOp {

    protected int[] dimension;

    public BaseBroadcastOp(INDArray x, INDArray y, INDArray z, int...dimension) {
        super(x,y,z,x.length());
        this.dimension = dimension;
        for(int i = 0; i < dimension.length; i++)
            if(dimension[i] < 0)
                dimension[i] += x.rank();
    }

    public BaseBroadcastOp(INDArray x, INDArray y, INDArray z, int n) {
        this(x,y,z,new int[]{n});
    }

    @Override
    public int broadcastLength() {
        if(y == null)
            throw new IllegalStateException("Unable to get broad cast length for y, no y specified");
        return y.length();
    }

    @Override
    public int[] broadcastShape() {
        if(y == null)
            throw new IllegalStateException("Unable to get broad cast shape for y, no y specified");
        return y.shape();
    }

    @Override
    public int[] getDimension(){
        return dimension;
    }

    @Override
    public void setDimension(int...dimension){
        this.dimension = dimension;
    }

    @Override
    public Op opForDimension(int index, int dimension){
        throw new UnsupportedOperationException("opForDimension not supported for BroadcastOps");
    }

    @Override
    public Op opForDimension(int index, int... dimension) {
        throw new UnsupportedOperationException("opForDimension not supported for BroadcastOps");
    }
}