org.nd4j.linalg.api.ops.exception.BlasOpErrorMessage Maven / Gradle / Ivy

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.exception;

import org.nd4j.linalg.api.blas.BlasBufferUtil;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.api.ops.Op;

import java.io.Serializable;
import java.util.Arrays;

/**
 * @author Adam Gibson
 */
public class BlasOpErrorMessage implements Serializable {
    private Op op;

    public BlasOpErrorMessage(Op op) {
        this.op = op;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder().append("Op " + op.name() + " of length " + op.n())
                        .append(" will fail with x of " + shapeInfo(op.x()));
        if (op.y() != null) {
            sb.append(" y of " + shapeInfo(op.y()));
        }

        sb.append(" and z of " + op.z());
        return sb.toString();
    }

    private String shapeInfo(INDArray arr) {
        return Arrays.toString(arr.shape()) + " and stride " + Arrays.toString(arr.stride()) + " and offset "
                        + arr.offset() + " and blas stride of " + BlasBufferUtil.getBlasStride(arr);
    }


}