Witllm/tools/test.py

7 lines
126 B
Python
Raw Normal View History

2023-12-21 19:52:19 +08:00
import show
import torch
radata = torch.randn(8192, 128)
show.DumpTensorToImage(radata, "test.png", autoPad=True,scale=0.2)