查询torch参数维度

在 PyTorch 中，你可以使用`torch.Tensor.size()`方法或者`torch.Tensor.shape`属性获取张量的维度信息。对于一个查询参数张量，它通常是一个形状为`(batch_size, query_dim)`的二维张量，其中`batch_size`表示输入数据的批处理大小，`query_dim`表示查询向量的维度。例如，如果你有一个名为`query`的查询参数张量，你可以使用以下代码获取其维度信息：

batch_size, query_dim = query.size()
print("batch_size:", batch_size)
print("query_dim:", query_dim)

或者

batch_size, query_dim = query.shape
print("batch_size:", batch_size)
print("query_dim:", query_dim)

其中`size()`方法返回一个元组，包含张量在每个维度上的大小，而`shape`属性返回一个包含张量在每个维度上大小的元组。