有时候需要可视化观察一下模型中间输出的结果,写个博客简单记录一下,方便后面直接粘贴复用
import numpy as np
import matplotlib.pyplot as plt
#xq_out.shape [B,H,W]
normalized_tensor = (xq_out[0] - xq_out[0].min()) /
有时候需要可视化观察一下模型中间输出的结果,写个博客简单记录一下,方便后面直接粘贴复用
import numpy as np
import matplotlib.pyplot as plt
#xq_out.shape [B,H,W]
normalized_tensor = (xq_out[0] - xq_out[0].min()) /