@apjs/tensor/src/creation/like.ts

A library with a set of functions to facilitate the use of the basic operations of linear algebra

import { Tensor, TensorLike } from "../class"
import { ensureTensor } from "../utils"
import { fill } from "./fill"


export const fillLike = (tensor: Tensor | TensorLike, number: number) => {
    let internTensor = ensureTensor(tensor)

    return fill(internTensor.shape, number)
}

export const onesLike = (tensor: Tensor | TensorLike) => {

    return fillLike(tensor, 1)
}

export const zerosLike = (tensor: Tensor | TensorLike) => {

    return fillLike(tensor, 0)
}