Nemo's Parental Misguidance
Share your code. npm Orgs help your team discover, share, and reuse code. Create a free org »

ndarray-gradient

1.0.0 • Public • Published

ndarray-gradient

Computes the gradient of an ndarray using a 2-point central finite difference template.

Example

var pack = require('ndarray-pack')
var pool = require('ndarray-scratch')
var grad = require('ndarray-gradient')
var show = require('ndarray-show')
 
var X = pack([[0, 0, 0],
              [0, 1, 0],
              [0, 0, 0]])
 
//Compute gradient of X
var dX = grad(pool.zero([3,3,2]), X)
 
console.log('grad(X) = \n', show(dX))

Output:

grad(X) =
   0.000    0.000    0.000
  -0.500    0.000    0.500
   0.000    0.000    0.000

   0.000   -0.500    0.000
   0.000    0.000    0.000
   0.000    0.500    0.000

Install

npm install ndarray-gradient

API

require('ndarray-gradient')(dst, src[, bc])

Computes the gradient of src storing the result into dst.

  • dst is an array of gradient values. The shape of dst must be the shape of src with one additional dimension for the components of the gradient

  • src is the array to differentiate

  • bc is an array of boundary conditions. The boundary conditions are encoded as string values and must be one of the following values:

    • 'clamp' (Default) clamp boundary edges to boundary
    • 'mirror' mirror values across the boundary
    • 'wrap' wrap values across boundary

Returns dst

Credits

(c) 2014 Mikola Lysenko. MIT License

install

npm i ndarray-gradient

Downloadsweekly downloads

1,782

version

1.0.0

license

MIT

homepage

github.com

repository

Gitgithub

last publish

collaborators

  • avatar
  • avatar
  • avatar
  • avatar
  • avatar
  • avatar
Report a vulnerability