Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
breeze.linalg.Transpose.scala Maven / Gradle / Ivy
package breeze.linalg
import breeze.linalg.operators._
import breeze.macros.expand
import breeze.linalg.support.{CanSlice, CanTranspose}
import breeze.generic.UFunc
/**
* TODO
*
* @author dlwh
**/
final case class Transpose[+T](val inner: T) extends NumericOps[Transpose[T]] {
def repr = this
}
object Transpose extends TransposeLowPrio {
implicit def canUntranspose[T]: CanTranspose[Transpose[T], T] = {
new CanTranspose[Transpose[T], T] {
def apply(from: Transpose[T]): T = from.inner
}
}
implicit def transTimesNormalFromDot[T, U, R](implicit dot: OpMulInner.Impl2[T, U, R]): OpMulMatrix.Impl2[Transpose[T], U, R] = {
new OpMulMatrix.Impl2[Transpose[T], U, R] {
def apply(v: Transpose[T], v2: U): R = {
dot(v.inner, v2)
}
}
}
implicit def transMulMatrix[T, U, R, RT](implicit op: OpMulMatrix.Impl2[T, U, R],
canTranspose: CanTranspose[R, RT]): OpMulMatrix.Impl2[Transpose[U], Transpose[T], RT] = {
new OpMulMatrix.Impl2[Transpose[U], Transpose[T], RT] {
def apply(v: Transpose[U], v2: Transpose[T]): RT = canTranspose(op(v2.inner, v.inner))
}
}
}
trait TransposeLowPrio {
implicit def liftOps[Op, T, U, R, RT](implicit op: UFunc.UImpl2[Op, T, U, R], canTranspose: CanTranspose[R, RT]):UFunc.UImpl2[Op, Transpose[T], Transpose[U], RT] = {
new UFunc.UImpl2[Op, Transpose[T], Transpose[U], RT] {
def apply(a: Transpose[T], b: Transpose[U]) = {
canTranspose(op(a.inner,b.inner))
}
}
}
implicit def liftInPlaceOps[Op, T, U](implicit op: UFunc.InPlaceImpl2[Op, T, U]):UFunc.InPlaceImpl2[Op, Transpose[T], Transpose[U]] = {
new UFunc.InPlaceImpl2[Op, Transpose[T], Transpose[U]] {
def apply(a: Transpose[T], b: Transpose[U]) {
op(a.inner,b.inner)
}
}
}
implicit class LiftApply[K, T](_trans: Transpose[Tensor[K, T]]) {
def apply(i: K):T = _trans.inner(i)
}
// TODO: make CanSlice a UFunc
implicit def liftSlice[Op, T, S, U, UT](implicit op: CanSlice[T, S, U], trans: CanTranspose[U, UT]): CanSlice[Transpose[T], S, UT] = {
new CanSlice[Transpose[T], S, UT] {
override def apply(from: Transpose[T], slice: S): UT = {
op(from.inner, slice).t
}
}
}
implicit def liftUFunc[Op, T, U, UT](implicit op: UFunc.UImpl[Op, T, U], trans: CanTranspose[U, UT]):UFunc.UImpl[Op, Transpose[T], UT] = {
new UFunc.UImpl[Op, Transpose[T], UT] {
override def apply(v: Transpose[T]): UT = trans(op(v.inner))
}
}
implicit def liftInPlace[Op, T, U](implicit op: UFunc.InPlaceImpl[Op, T]):UFunc.InPlaceImpl[Op, Transpose[T]] = {
new UFunc.InPlaceImpl[Op, Transpose[T]] {
override def apply(v: Transpose[T]) = op(v.inner)
}
}
implicit def liftUFunc3_1[Op, T, T2, T3, U, UT](implicit op: UFunc.UImpl3[Op, T, T2, T3, U],
trans: CanTranspose[U, UT]):UFunc.UImpl3[Op, Transpose[T], T2, T3, UT] = {
new UFunc.UImpl3[Op, Transpose[T], T2, T3, UT] {
override def apply(v: Transpose[T], v2: T2, v3: T3): UT = {
trans(op(v.inner, v2, v3))
}
}
}
implicit def liftUFuncInplace3_1[Op, T, T2, T3](implicit op: UFunc.InPlaceImpl3[Op, T, T2, T3]):UFunc.InPlaceImpl3[Op, Transpose[T], T2, T3] = {
new UFunc.InPlaceImpl3[Op, Transpose[T], T2, T3] {
override def apply(v: Transpose[T], v2: T2, v3: T3) = {
op(v.inner, v2, v3)
}
}
}
}
