convolution.py
.numpy
is useful for handling vectors and matrices.main()
should return the object of type pyplot
used for plotting. For example:
import matplotlib.pyplot as plt
import numpy as np
...
def main():
fig = plt.figure(figsize=(18, 9))
...
return fig
if __name__ == "__main__":
main()