Witllm/tools/test.py

14 lines
274 B
Python

import show
import torch
# radata = torch.randn(8192, 128)
# show.DumpTensorToImage(radata, "test.png", autoPad=True,scale=0.2)
radata = torch.randn(127)
show.DumpTensorToImage(radata, "test.png")
radata = torch.randn(127, 127)
show.DumpTensorToLog(radata, "test.log")