pytorch怎么打印tensor的值

小億
371
2024-03-29 13:11:09

在PyTorch中，可以使用print()函數(shù)打印Tensor的值。例如：

import torch

# 創(chuàng)建一個(gè)Tensor
x = torch.tensor([1, 2, 3, 4, 5])

# 打印Tensor的值
print(x)

輸出結(jié)果會(huì)顯示Tensor的值：

tensor([1, 2, 3, 4, 5])

如果想要將Tensor的值轉(zhuǎn)換為Numpy數(shù)組進(jìn)行打印，可以使用.numpy()方法。例如：

import torch

# 創(chuàng)建一個(gè)Tensor
x = torch.tensor([1, 2, 3, 4, 5])

# 打印Tensor的值
print(x.numpy())

這樣就會(huì)將Tensor的值轉(zhuǎn)換為Numpy數(shù)組進(jìn)行打印。

0 踩