pytorch学习1:tensor的操作
tensor操作中的api和numpy非常相似
1、基本操作
x = torch.ones(2,2)
print(x)
x.type()
# 将其转换为整型
x = x.long()
print(x)
# 再将其转回float
x = x.float()
print(x)
x = torch.randn(4,3)
print(x)
# 沿着行取最大值
max_value, max_idx...
原创
2019-01-16 14:22:45 ·
756 阅读 ·
0 评论