Function to index tensors columns
tf_stride_cols(A, start, end = NULL)
A
tensor
start
first index
end
last index (equals start index if NULL)
sliced tensor