Eu queria criar um mapa de calor de uma matriz de densidade de probabilidade usando plotly.

import numpy as np
from plotly.offline import download_plotlyjs, init_notebook_mode, plot
import plotly.graph_objs as go

probability_matrix = np.loadtxt("/path/to/file")
trace = go.Heatmap(z = probability_matrix)
data=[trace]
plot(data, filename='basic-heatmap')

Isso me dá uma imagem como esta:

Quero suavizar a cor dos quadrados para que a transição entre os quadrados adjacentes na imagem seja um pouco "mais suave". Fiquei me perguntando se existe uma maneira de fazer isso, sem redimensionar manualmente a matriz usando interpolação.