Corrected mistakes in transform

Author: theogf

src/transform/lowranktransform.jl

@@ -6,10 +6,13 @@ Base.size(tr::LowRankTransform,i::Int) = size(tr.proj,i)
 Base.size(tr::LowRankTransform) = size(tr.proj)
 
 function transform(t::LowRankTransform,X::AbstractMatrix{<:Real},obsdim::Int)
-    @boundscheck if size(t,2) != size(X,1)
-        throw(DimensionMismatch("The projection matrix has size $(size(t)) and cannot be used on X with dimensions $(size(X))"))
-    end
-    _transform(t,X,obsdim)
+    @boundscheck size(t,2) != size(X,feature_dim(obsdim)) ?
+        throw(DimensionMismatch("The projection matrix has size $(size(t)) and cannot be used on X with dimensions $(size(X))")) : nothing
+    @inbounds _transform(t,X,obsdim)
 end
-_transform(t::LowRankTransform,x::AbstractVector{<:Real}) = t.proj * x
-_transform(t::LowRankTransform,X::AbstractMatrix{<:Real},obsdim::Int) = t.proj * X
+function transform(t::LowRankTransform,x::AbstractVector{<:Real})
+    @assert size(t,2) == length(x) "Vector has wrong dimensions"
+    t.proj*X
+end
+
+_transform(t::LowRankTransform,X::AbstractVecOrMat{<:Real},obsdim::Int) = obsdim == 2 ? t.proj * X : X * t.proj'

src/transform/transform.jl

@@ -31,15 +31,15 @@ Base.:∘(tc::TransformChain,t::Transform) = TransformChain(vcat(t,tc.transforms
 """
     FunctionTransform
 
-    Take a function `f` as an argument which is going to act on the matrix as a whole.
-    Make sure that `f` is supposed to act on a matrix by eventually using broadcasting
+    Take a function `f` as an argument which is going to act on each vector individually.
+    Make sure that `f` is supposed to act on a vector by eventually using broadcasting
     For example `f(x)=sin(x)` -> `f(x)=sin.(x)`
 """
 struct FunctionTransform{F} <: Transform
     f::F
 end
 
-transform(t::FunctionTransform,X::T,obsdim::Int=1) where {T} = t.f(X)
+transform(t::FunctionTransform,X::T,obsdim::Int=1) where {T} = mapslices(t.f,X,obsdim) 
 
 
 struct IdentityTransform <: Transform end