TensorCast.jl
TensorCast.jl copied to clipboard
mcabbott
It slices, it dices, it splices!
I am getting a world age bug when running TensorCast in JuliaPy, but not when using it normally. If I run an example function such as ``` function testfunc() V...
I wanted to reduce into a 2-vector by using the indexes stored in `ixs`. Currently it seems to ignore `ixs`. What do you think about this idea? ```jl using TensorCast...
Would be neat if this worked (from [this thread](https://discourse.julialang.org/t/inner-repeat-without-allocations/72204)): ```julia julia> X = randn(1000, 1000); julia> Y = zeros(2000, 2000); julia> Y .= repeat(X, inner=(2,2)); julia> @cast Y[(a,b), (c,d)] =...
This should work: ```julia julia> using TensorCast, Zygote julia> f1(x,y) = [x^2, x*y, y^2]; julia> x = 1:2; y = 1:4; julia> @cast out[i,j,k] := f1(x[i], y[j])[k] 2×4×3 transmute(stack(::Matrix{Vector{Int64}}), (2,...
``` julia> M = Array(reshape(1:12, 3,4)) 3×4 Matrix{Int64}: 1 4 7 10 2 5 8 11 3 6 9 12 julia> @cast R[r,(n,c)] := M[r,c]^2 (n in 1:3) ERROR: LoadError:...
On master the following syntax is understood: ``` julia> using TensorCast, ImageCore julia> mat = rand(1:10, 4, 3) ./ 10; julia> @cast _[k] := RGB(mat[k,:]...) 4-element Array{RGB{Float64},1} with eltype RGB{Float64}:...
This `@capture` replacement is 100x quicker, but fixing one pattern doesn't improve the package startup speed much.
I'm trying to slice-broadcast a function that has multiple return values. Ideally I'd collect them into separate arrays. Is this possible witih TensorCast? Here's a MWE of basically what I'm...
I was asked to make this an issue, but I'm not sure there's anything to be done about it. TensorCast uses scalar indexing when a shared index changes position, which...
The following: ``` using TensorCast, CUDA; CUDA.allowscalar(false) C = cu(ones(10,2)) L = cu(ones(10,3)) @reduce D[m,a] := sum(p) C[p,a] + L[p,m] ``` gives a `scalar getindex is disallowed` error. But using...
