Join GitHub today
GitHub is home to over 36 million developers working together to host and review code, manage projects, and build software together.Sign up
libnd4j: Add linspace op with dynamic array input args #6723
We've had a linspace op forever, but we need a version that takes INDArray arguments for the inputs (start, stop, num params) instead of the fixed "double" arguments. Required for TF import. Will also allow us to have 'dynamic' linspace in graphs (i.e., variable start/stop/num between runs) instead of being restricted to fixed args.
Not a super high high priority as I can (hopefully) workaround this for now (but it won't work for libnd4j export + execution).