打印张量的前几行 python

假设你已经有一个张量（tensor）了，可以使用`print`函数和切片（slicing）操作来打印张量的前几行。下面是一个示例代码：

import torch

# 创建一个大小为(5, 3)的张量
tensor = torch.randn(5, 3)

# 打印张量的前两行
print(tensor[:2])

输出结果类似于：

tensor([[-0.4299, -0.1263, -0.3152],
        [ 0.1183, -0.3396,  0.8721]])

在这个例子中，我们使用了切片操作`[:2]`来获取张量的前两行。你可以根据需要修改切片来打印更多或更少的行。