/batch-matmul-tensorflow

TensorFlow 2.0 Implementation for Batch Matrix Multiplication

Primary LanguageJupyter Notebook

Batch Matrix Multiplication of TensorFlow

This repository is for the TensorFlow implementation of batch matrix multiplication. This repository provides a simple test bench Jupyter Notebook for verification of exact working and broadcasting. If you want to use this batch matrix multiplication operator, check how to use the operator on the test bench notebook and use freely.

Prerequisite for Test Bench Jupyter Notebook

  • Python 3.5 or higher versions
  • Jupyter Notebook version 1.0.0 or higher versions
  • Numpy version 1.16.4 or higher versions
  • TensorFlow version 1.14.0 or higher versions