learnergy.visual.tensor¶
Tensor-related visualization.
- learnergy.visual.tensor.save_tensor(tensor: torch.Tensor, output_path: str)¶
Saves a tensor in grayscale mode using Matplotlib.
- Parameters
tensor – An input tensor to be saved.
output_path – An outputh path to save the tensor.
- learnergy.visual.tensor.show_tensor(tensor: torch.Tensor)¶
Plots a tensor in grayscale mode using Matplotlib.
- Parameters
tensor – An input tensor to be plotted.