class torch.nn.Embedding
store word embeddings and retrieve them using indices.
| Parameters: |
- num_embeddings (int) – size of the dictionary of embeddings
- embedding_dim (int) – the size of each embedding vector
| Variables: |
weight (Tensor) – the learnable weights of the module of shape (num_embeddings, embedding_dim)
torch.tensor
| Parameters: |
-
data (array_like) – Initial data for the tensor. Can be a list, tuple, NumPy
ndarray, scalar, and other types. -
dtype (
torch.dtype, optional) – the desired data type of returned tensor.
class torch.nn.MarginRankingLoss
transE中标准用法
torch.abs
Computes the element-wise absolute value of the given input tensor.
返回的是一个tensor。
view(*args)
Returns a new tensor with the same data as the self tensor but of a different size.
torch.mean()
Returns the mean value of all elements in the input tensor.
torch.mean(input, 1)
对每一行求平均
