tfjs-core
tfjs-core copied to clipboard
tf.linalg.bandPart
This is an attempt to further break down/simplify PR#1366.
(Matrix) band part is required for the backpropagation of many linear operations. Implementation should behave the same way as in Python/C/C++ Tensorflow.
@dsmilkov PTAL