torch.ones(2,1)解释

建立一个tensor张量，维度是2行1列。

print(torch.ones(2,1))
tensor([[1.],
[1.]])

posted @ 2021-06-25 22:44  祥瑞哈哈哈  阅读(640)  评论(0)  编辑  收藏  举报