创建tensor
y = torch.zeros(2, 3, 4, dtype=torch.int32)
print(y)
# 输出:
# tensor([[[0, 0, 0, 0],
# [0, 0, 0, 0],
# [0, 0, 0, 0]],
#
# [[0, 0, 0, 0],
# [0, 0, 0, 0],
# [0, 0, 0, 0]]], dtype=torch.int32)
创建tensor
y = torch.zeros(2, 3, 4, dtype=torch.int32)
print(y)
# 输出:
# tensor([[[0, 0, 0, 0],
# [0, 0, 0, 0],
# [0, 0, 0, 0]],
#
# [[0, 0, 0, 0],
# [0, 0, 0, 0],
# [0, 0, 0, 0]]], dtype=torch.int32)