com.expleague.ml.models.gpf.Tensor4 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jmll
Various ML methods implemented by myself and my students
There is a newer version: 1.4.9
package com.expleague.ml.models.gpf;

import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.math.vectors.impl.vectors.ArrayVec;

/**
 * User: irlab
 * Date: 16.05.14
 */
public class Tensor4 {
  public final ArrayVec vec;
  public final int dim1;
  public final int dim2;
  public final int dim3;
  public final int dim4;

  public Tensor4(final int dim1, final int dim2, final int dim3, final int dim4) {
    this.dim1 = dim1;
    this.dim2 = dim2;
    this.dim3 = dim3;
    this.dim4 = dim4;
    vec = new ArrayVec(dim1 * dim2 * dim3 * dim4);
  }

  private int index(final int i, final int j, final int k, final int l) {
    assert(0 <= i && i < dim1);
    assert(0 <= j && j < dim2);
    assert(0 <= k && k < dim3);
    assert(0 <= l && l < dim4);
    return dim4 * (dim3 * (dim2 * i + j) + k) + l;
  }

  public double get(final int i, final int j, final int k, final int l) {
    return vec.get(index(i, j, k, l));
  }

  public Tensor4 set(final int i, final int j, final int k, final int l, final double val) {
    vec.set(index(i, j, k, l), val);
    return this;
  }

  public ArrayVec getRow(final int i, final int j, final int k) {
    return vec.sub(index(i, j, k, 0), dim4);
  }

  public Tensor4 setRow(final int i, final int j, final int k, final Vec val) {
    if (val.dim() != dim4)
      throw new IllegalArgumentException("val.xdim() != dim4, val.xdim() = " + val.dim() + ", dim4 = " + dim4);
    final int index = index(i, j, k, 0);
    for (int l = 0; l < dim4; l++)
      vec.set(index + l, val.get(l));
    return this;
  }

  public Tensor4 adjust(final int i, final int j, final int k, final int l, final double increment) {
    vec.adjust(index(i, j, k, l), increment);
    return this;
  }

  public String toString() {
    final StringBuilder builder = new StringBuilder();
    for (int i = 0; i < dim1; i++) {
      for (int j = 0; j < dim2; j++) {
        for (int k = 0; k < dim3; k++) {
          for (int l = 0; l < dim4; l++) {
            builder.append(l > 0 ? "\t" : "");
            builder.append(get(i, j, k, l));
          }
          builder.append('\n');
        }
        builder.append('\n');
      }
      builder.append('\n');
    }
    return builder.toString();
  }

  public double[] toArray() {
    return vec.toArray();
  }

  public boolean equals(final Object o) {
    return o instanceof Tensor3 && (((Tensor3)o).dim1 == dim1) && (((Tensor3)o).dim2 == dim2) && ((Tensor3)o).vec.equals(vec);
  }

  public int hashCode() {
    return (vec.hashCode() << 1) + dim2 * dim3 * dim4;
  }
}